Home
last modified time | relevance | path

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

/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()
333 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in RunDotOutputFusion()
Dcpu_compiler.cc394 LayoutAssignment::InstructionCanChangeLayout, target_machine_features); in RunHloPassesThroughLayoutAssn()
420 LayoutAssignment::InstructionCanChangeLayout); in RunHloPassesAfterLayoutAssn()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc80 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
169 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
240 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
322 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
359 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
397 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
424 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in TEST_F()
Dgpu_layout_assignment.h30 class GpuLayoutAssignment : public LayoutAssignment {
36 : LayoutAssignment(entry_computation_layout, in GpuLayoutAssignment()
Dgpu_layout_assignment.cc356 return LayoutAssignment::PropagateOperandConstraint(layout_constraint, in PropagateOperandConstraint()
397 return LayoutAssignment::PropagateBufferConstraint(buffer_constraint, in PropagateBufferConstraint()
Dgpu_compiler.cc301 LayoutAssignment::InstructionCanChangeLayout, stream_exec); in OptimizeHloModule()
319 LayoutAssignment::InstructionCanChangeLayout); in OptimizeHloModule()
371 LayoutAssignment::InstructionCanChangeLayout); in PrepareHloModuleForIrEmitting()
411 LayoutAssignment::InstructionCanChangeLayout); in OptimizeHloPostLayoutAssignment()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc401 Status LayoutAssignment::BuildHostChannelConstraints( in BuildHostChannelConstraints()
444 Status LayoutAssignment::AddMandatoryConstraints( in AddMandatoryConstraints()
799 StatusOr<HloInstruction*> LayoutAssignment::CreateCopyWithNewLayout( in CreateCopyWithNewLayout()
859 Status LayoutAssignment::CopyOperandIfLayoutsDiffer( in CopyOperandIfLayoutsDiffer()
908 void LayoutAssignment::SetupCopiedInstruction(const HloInstruction& instruction, in SetupCopiedInstruction()
930 Status LayoutAssignment::CheckLayouts(HloModule* module) { in CheckLayouts()
1024 LayoutAssignment::LayoutAssignment( in LayoutAssignment() function in xla::LayoutAssignment
1044 std::unique_ptr<Layout> LayoutAssignment::ChooseOperandLayoutFromOutputLayout( in ChooseOperandLayoutFromOutputLayout()
1111 std::unique_ptr<Layout> LayoutAssignment::ChooseOutputLayoutFromOperandLayout( in ChooseOutputLayoutFromOperandLayout()
1172 Status LayoutAssignment::PropagateConstraints(LayoutConstraints* constraints) { in PropagateConstraints()
[all …]
Dlayout_assignment.h286 class LayoutAssignment : public HloModulePass {
303 explicit LayoutAssignment(
308 ~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()
829 LayoutAssignment layout_assignment(&computation_layout); in TEST_F()
1304 LayoutAssignment layout_assignment( in AssignLayoutsToComputation()
1306 LayoutAssignment::InstructionCanChangeLayout, channel_constraints); in AssignLayoutsToComputation()
Dhlo_verifier_test.cc66 LayoutAssignment::InstructionCanChangeLayout) {} in HloVerifierTestLayoutSensitive()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc89 pipeline.AddPass<LayoutAssignment>( in RunHloOptimization()
91 LayoutAssignment::InstructionCanChangeLayout); in RunHloOptimization()