Home
last modified time | relevance | path

Searched refs:LayoutAssignment (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc391 Status LayoutAssignment::BuildHostChannelConstraints( in BuildHostChannelConstraints()
428 Status LayoutAssignment::AddMandatoryConstraints( in AddMandatoryConstraints()
772 StatusOr<HloInstruction*> LayoutAssignment::CreateCopyWithNewLayout( in CreateCopyWithNewLayout()
831 Status LayoutAssignment::CopyOperandIfLayoutsDiffer( in CopyOperandIfLayoutsDiffer()
855 void LayoutAssignment::SetupCopiedInstruction(const HloInstruction& instruction, in SetupCopiedInstruction()
877 Status LayoutAssignment::CheckLayouts(HloModule* module) { in CheckLayouts()
967 LayoutAssignment::LayoutAssignment( in LayoutAssignment() function in xla::LayoutAssignment
987 std::unique_ptr<Layout> LayoutAssignment::ChooseOperandLayoutFromOutputLayout( in ChooseOperandLayoutFromOutputLayout()
1053 std::unique_ptr<Layout> LayoutAssignment::ChooseOutputLayoutFromOperandLayout( in ChooseOutputLayoutFromOperandLayout()
1114 Status LayoutAssignment::PropagateConstraints(LayoutConstraints* constraints) { in PropagateConstraints()
[all …]
Dlayout_assignment.h284 class LayoutAssignment : public HloModulePass {
301 explicit LayoutAssignment(
306 ~LayoutAssignment() override {} in ~LayoutAssignment()
Dlayout_assignment_test.cc56 LayoutAssignment layout_assignment( in AssignLayouts()
57 entry_computation_layout, LayoutAssignment::InstructionCanChangeLayout, in AssignLayouts()
317 LayoutAssignment layout_assignment(&computation_layout); in TEST_F()
514 class OperandsMustBeTheSameLayoutAssignment : public LayoutAssignment {
518 : LayoutAssignment(entry_computation_layout) {} in OperandsMustBeTheSameLayoutAssignment()
832 LayoutAssignment layout_assignment(&computation_layout); in TEST_F()
1310 LayoutAssignment layout_assignment( in AssignLayoutsToComputation()
1312 LayoutAssignment::InstructionCanChangeLayout, channel_constraints); in AssignLayoutsToComputation()
Dhlo_verifier_test.cc65 LayoutAssignment::InstructionCanChangeLayout) {} in HloVerifierTestLayoutSensitive()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.h29 class CpuLayoutAssignment : public LayoutAssignment {
36 : LayoutAssignment(entry_computation_layout, in CpuLayoutAssignment()
Dcpu_layout_assignment_test.cc58 entry_computation_layout, LayoutAssignment::InstructionCanChangeLayout, in AssignLayouts()
326 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in RunDotOutputFusion()
Dcpu_compiler.cc325 LayoutAssignment::InstructionCanChangeLayout, target_machine_features); in RunHloPassesThroughLayoutAssn()
356 LayoutAssignment::InstructionCanChangeLayout); in RunHloPassesAfterLayoutAssn()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc78 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
167 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
238 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
320 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
357 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
395 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
Dgpu_layout_assignment.h30 class GpuLayoutAssignment : public LayoutAssignment {
36 : LayoutAssignment(entry_computation_layout, in GpuLayoutAssignment()
Dnvptx_compiler.cc300 LayoutAssignment::InstructionCanChangeLayout, stream_exec); in OptimizeHloModule()
311 LayoutAssignment::InstructionCanChangeLayout); in OptimizeHloModule()
364 LayoutAssignment::InstructionCanChangeLayout); in OptimizeHloModule()
379 LayoutAssignment::InstructionCanChangeLayout); in OptimizeHloModule()
410 LayoutAssignment::InstructionCanChangeLayout); in PrepareHloModuleForIrEmitting()
Dgpu_layout_assignment.cc323 return LayoutAssignment::PropagateOperandConstraint(layout_constraint, in PropagateOperandConstraint()
364 return LayoutAssignment::PropagateBufferConstraint(buffer_constraint, in PropagateBufferConstraint()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc86 pipeline.AddPass<LayoutAssignment>( in RunHloOptimization()
88 LayoutAssignment::InstructionCanChangeLayout); in RunHloOptimization()