Searched refs:ChannelLayoutConstraints (Results 1 – 3 of 3) sorted by relevance
239 class ChannelLayoutConstraints {242 ChannelLayoutConstraints() {} in ChannelLayoutConstraints() function305 ChannelLayoutConstraints* channel_constraints = nullptr);380 ChannelLayoutConstraints* channel_constraints,401 ChannelLayoutConstraints* channel_constraints);488 ChannelLayoutConstraints* channel_constraints);513 ChannelLayoutConstraints* channel_layout_constraints_ = nullptr;518 ChannelLayoutConstraints channel_constraints_;522 ChannelLayoutConstraints host_channel_constraints_;
55 ChannelLayoutConstraints* channel_constraints = nullptr) { in AssignLayouts()872 ChannelLayoutConstraints channel_constraints; in TEST_F()916 ChannelLayoutConstraints channel_constraints; in TEST_F()1304 HloModule* m, ChannelLayoutConstraints* channel_constraints = nullptr) { in AssignLayoutsToComputation()
430 ChannelLayoutConstraints* channel_constraints, HloComputation* computation, in AddMandatoryConstraints()971 ChannelLayoutConstraints* channel_constraints) in LayoutAssignment()1703 ChannelLayoutConstraints* channel_constraints) { in RunOnComputation()1810 ChannelLayoutConstraints* channel_constraints) { in ConstrainChannelLayouts()