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
168 GpuLayoutAssignment layout_assignment( in TEST_F() local
239 GpuLayoutAssignment layout_assignment( in TEST_F() local
321 GpuLayoutAssignment layout_assignment( in TEST_F() local
358 GpuLayoutAssignment layout_assignment( in TEST_F() local
396 GpuLayoutAssignment layout_assignment( in TEST_F() local
423 GpuLayoutAssignment layout_assignment( in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgpu_fusion.cc131 static PassRegistration<GpuOpFusionPass> layout_assignment( variable
Dlayout_optimization.cc483 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
1304 LayoutAssignment layout_assignment( in AssignLayoutsToComputation() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc57 cpu::CpuLayoutAssignment layout_assignment( in AssignLayouts() local
332 cpu::CpuLayoutAssignment layout_assignment( in RunDotOutputFusion() local