Searched refs:channel_constraints (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.h | 307 ChannelLayoutConstraints* channel_constraints = nullptr);
389 ChannelLayoutConstraints* channel_constraints,
409 ChannelLayoutConstraints* channel_constraints);
496 ChannelLayoutConstraints* channel_constraints);
|
| D | layout_assignment_test.cc | 55 ChannelLayoutConstraints* channel_constraints = nullptr) { in AssignLayouts() argument
58 /*channel_constraints=*/channel_constraints); in AssignLayouts()
869 ChannelLayoutConstraints channel_constraints; in TEST_F() local
870 AssignLayouts(m.get(), &computation_layout, &channel_constraints); in TEST_F()
910 ChannelLayoutConstraints channel_constraints; in TEST_F() local
911 AssignLayouts(m.get(), &computation_layout, &channel_constraints); in TEST_F()
1298 HloModule* m, ChannelLayoutConstraints* channel_constraints = nullptr) { in AssignLayoutsToComputation() argument
1306 LayoutAssignment::InstructionCanChangeLayout, channel_constraints); in AssignLayoutsToComputation()
1379 ChannelLayoutConstraints channel_constraints; in TEST_F() local
1380 AssignLayouts(m.get(), &computation_layout, &channel_constraints); in TEST_F()
[all …]
|
| D | layout_assignment.cc | 446 ChannelLayoutConstraints* channel_constraints, HloComputation* computation, in AddMandatoryConstraints() argument
453 : channel_constraints; in AddMandatoryConstraints()
1028 ChannelLayoutConstraints* channel_constraints) in LayoutAssignment() argument
1032 channel_layout_constraints_(channel_constraints), in LayoutAssignment()
1843 ChannelLayoutConstraints* channel_constraints) { in RunOnComputation() argument
1872 computation_layout, channel_constraints, computation, &constraints)); in RunOnComputation()
1929 if (channel_constraints != nullptr) { in RunOnComputation()
1931 ConstrainChannelLayouts(computation, channel_constraints)); in RunOnComputation()
1972 ChannelLayoutConstraints* channel_constraints) { in ConstrainChannelLayouts() argument
1975 : channel_constraints; in ConstrainChannelLayouts()
|