Searched refs:channel_constraints (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.h | 305 ChannelLayoutConstraints* channel_constraints = nullptr);
380 ChannelLayoutConstraints* channel_constraints,
401 ChannelLayoutConstraints* channel_constraints);
488 ChannelLayoutConstraints* channel_constraints);
|
| D | layout_assignment_test.cc | 55 ChannelLayoutConstraints* channel_constraints = nullptr) { in AssignLayouts() argument
58 /*channel_constraints=*/channel_constraints); in AssignLayouts()
872 ChannelLayoutConstraints channel_constraints; in TEST_F() local
873 AssignLayouts(m.get(), &computation_layout, &channel_constraints); in TEST_F()
916 ChannelLayoutConstraints channel_constraints; in TEST_F() local
917 AssignLayouts(m.get(), &computation_layout, &channel_constraints); in TEST_F()
1304 HloModule* m, ChannelLayoutConstraints* channel_constraints = nullptr) { in AssignLayoutsToComputation() argument
1312 LayoutAssignment::InstructionCanChangeLayout, channel_constraints); in AssignLayoutsToComputation()
|
| D | layout_assignment.cc | 430 ChannelLayoutConstraints* channel_constraints, HloComputation* computation, in AddMandatoryConstraints() argument
437 : channel_constraints; in AddMandatoryConstraints()
971 ChannelLayoutConstraints* channel_constraints) in LayoutAssignment() argument
975 channel_layout_constraints_(channel_constraints), in LayoutAssignment()
1703 ChannelLayoutConstraints* channel_constraints) { in RunOnComputation() argument
1732 computation_layout, channel_constraints, computation, &constraints)); in RunOnComputation()
1789 if (channel_constraints != nullptr) { in RunOnComputation()
1791 ConstrainChannelLayouts(computation, channel_constraints)); in RunOnComputation()
1810 ChannelLayoutConstraints* channel_constraints) { in ConstrainChannelLayouts() argument
1813 : channel_constraints; in ConstrainChannelLayouts()
|