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()
|