Searched defs:channel_constraints (Results 1 – 4 of 4) sorted by relevance
55 ChannelLayoutConstraints* channel_constraints = nullptr) { in AssignLayouts()862 ChannelLayoutConstraints channel_constraints; in TEST_F() local903 ChannelLayoutConstraints channel_constraints; in TEST_F() local1334 HloModule* m, ChannelLayoutConstraints* channel_constraints = nullptr) { in AssignLayoutsToComputation()1414 ChannelLayoutConstraints channel_constraints; in TEST_F() local1451 ChannelLayoutConstraints channel_constraints; in TEST_F() local1530 ChannelLayoutConstraints channel_constraints; in TEST_F() local1597 ChannelLayoutConstraints channel_constraints; in TEST_F() local
575 ChannelLayoutConstraints* channel_constraints, in AddMandatoryConstraints()1242 ChannelLayoutConstraints* channel_constraints, in LayoutAssignment()2229 ChannelLayoutConstraints* channel_constraints) { in RunOnComputation()2311 ChannelLayoutConstraints* channel_constraints) { in ConstrainChannelLayouts()
35 : LayoutAssignment(entry_computation_layout, channel_constraints), in LayoutAssignment() argument
36 : LayoutAssignment(entry_computation_layout, channel_constraints), in LayoutAssignment() argument