Searched defs:layout_assignment (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_layout_assignment_test.cc | 82 GpuLayoutAssignment layout_assignment( in TEST_F() local
112 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
138 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
165 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
191 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
218 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
245 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
280 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
305 GpuLayoutAssignment layout_assignment(&computation_layout, in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment_test.cc | 56 LayoutAssignment layout_assignment( in AssignLayouts() local
283 LayoutAssignment layout_assignment(&computation_layout); in TEST_F() local
533 OperandsMustBeTheSameLayoutAssignment layout_assignment(&computation_layout); in TEST_F() local
805 LayoutAssignment layout_assignment(&computation_layout); in TEST_F() local
822 LayoutAssignment layout_assignment(&computation_layout); in TEST_F() local
1340 LayoutAssignment layout_assignment(m->mutable_entry_computation_layout(), in AssignLayoutsToComputation() local
1531 LayoutAssignment layout_assignment( in TEST_F() local
1598 LayoutAssignment layout_assignment( in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_layout_assignment_test.cc | 57 cpu::CpuLayoutAssignment layout_assignment(entry_computation_layout, in AssignLayouts() local
331 cpu::CpuLayoutAssignment layout_assignment(&computation_layout, in RunDotOutputFusion() local
|