Home
last modified time | relevance | path

Searched defs:layout_assignment (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc79 GpuLayoutAssignment layout_assignment( in TEST_F() local
167 GpuLayoutAssignment layout_assignment( in TEST_F() local
237 GpuLayoutAssignment layout_assignment( in TEST_F() local
318 GpuLayoutAssignment layout_assignment( in TEST_F() local
354 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
391 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
417 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgpu_fusion.cc132 static PassRegistration<GpuOpFusionPass> layout_assignment; variable
Dlayout_optimization.cc512 static PassRegistration<LayoutAssignmentPass> layout_assignment; variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc56 LayoutAssignment layout_assignment( in AssignLayouts() local
317 LayoutAssignment layout_assignment(&computation_layout); in TEST_F() local
570 OperandsMustBeTheSameLayoutAssignment layout_assignment(&computation_layout); in TEST_F() local
829 LayoutAssignment layout_assignment(&computation_layout); in TEST_F() local
1332 LayoutAssignment layout_assignment(m->mutable_entry_computation_layout(), in AssignLayoutsToComputation() local
1525 LayoutAssignment layout_assignment( in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc57 cpu::CpuLayoutAssignment layout_assignment(entry_computation_layout, in AssignLayouts() local
331 cpu::CpuLayoutAssignment layout_assignment(&computation_layout, in RunDotOutputFusion() local