Searched refs:LayoutAssignment (Results 1 – 10 of 10) sorted by relevance
439 Status LayoutAssignment::BuildHostChannelConstraints( in BuildHostChannelConstraints()510 Status LayoutAssignment::AddMandatoryConstraints( in AddMandatoryConstraints()886 StatusOr<HloInstruction*> LayoutAssignment::CreateCopyWithNewLayout( in CreateCopyWithNewLayout()946 Status LayoutAssignment::CopyOperandIfLayoutsDiffer( in CopyOperandIfLayoutsDiffer()996 void LayoutAssignment::SetupCopiedInstruction(const HloInstruction& instruction, in SetupCopiedInstruction()1018 Status LayoutAssignment::CheckLayouts(HloModule* module) { in CheckLayouts()1112 LayoutAssignment::LayoutAssignment( in LayoutAssignment() function in xla::LayoutAssignment1129 std::unique_ptr<Layout> LayoutAssignment::ChooseOperandLayoutFromOutputLayout( in ChooseOperandLayoutFromOutputLayout()1196 std::unique_ptr<Layout> LayoutAssignment::ChooseOutputLayoutFromOperandLayout( in ChooseOutputLayoutFromOperandLayout()1257 Status LayoutAssignment::PropagateConstraints(LayoutConstraints* constraints) { in PropagateConstraints()[all …]
306 class LayoutAssignment : public HloModulePass {318 explicit LayoutAssignment(322 ~LayoutAssignment() override {} in ~LayoutAssignment()
56 LayoutAssignment layout_assignment( in AssignLayouts()317 LayoutAssignment layout_assignment(&computation_layout); in TEST_F()514 class OperandsMustBeTheSameLayoutAssignment : public LayoutAssignment {518 : LayoutAssignment(entry_computation_layout) {} in OperandsMustBeTheSameLayoutAssignment()829 LayoutAssignment layout_assignment(&computation_layout); in TEST_F()1332 LayoutAssignment layout_assignment(m->mutable_entry_computation_layout(), in AssignLayoutsToComputation()1525 LayoutAssignment layout_assignment( in TEST_F()
66 LayoutAssignment::InstructionCanChangeLayout) {} in HloVerifierTestLayoutSensitive()
29 class CpuLayoutAssignment : public LayoutAssignment {34 : LayoutAssignment(entry_computation_layout), in CpuLayoutAssignment()
445 LayoutAssignment::InstructionCanChangeLayout); in RunHloPassesAfterLayoutAssn()
30 class GpuLayoutAssignment : public LayoutAssignment {36 : LayoutAssignment(entry_computation_layout, in LayoutAssignment() function
398 return LayoutAssignment::PropagateOperandConstraint(layout_constraint, in PropagateOperandConstraint()439 return LayoutAssignment::PropagateBufferConstraint(buffer_constraint, in PropagateBufferConstraint()
452 LayoutAssignment::InstructionCanChangeLayout); in OptimizeHloModule()518 LayoutAssignment::InstructionCanChangeLayout); in PrepareHloModuleForIrEmitting()543 LayoutAssignment::InstructionCanChangeLayout); in OptimizeHloPostLayoutAssignment()
91 pipeline.AddPass<LayoutAssignment>( in RunHloOptimization()