Searched refs:InstructionCanChangeLayout (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 78 &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()
|
| D | nvptx_compiler.cc | 300 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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.h | 304 instruction_can_change_layout_func = InstructionCanChangeLayout,
316 static bool InstructionCanChangeLayout(const HloInstruction* instruction);
|
| D | hlo_verifier_test.cc | 65 LayoutAssignment::InstructionCanChangeLayout) {} in HloVerifierTestLayoutSensitive()
|
| D | layout_assignment_test.cc | 57 entry_computation_layout, LayoutAssignment::InstructionCanChangeLayout, in AssignLayouts()
1312 LayoutAssignment::InstructionCanChangeLayout, channel_constraints); in AssignLayoutsToComputation()
|
| D | layout_assignment.cc | 1978 bool LayoutAssignment::InstructionCanChangeLayout( in InstructionCanChangeLayout() function in xla::LayoutAssignment
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 88 LayoutAssignment::InstructionCanChangeLayout); in RunHloOptimization()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment_test.cc | 58 entry_computation_layout, LayoutAssignment::InstructionCanChangeLayout, in AssignLayouts()
326 &computation_layout, LayoutAssignment::InstructionCanChangeLayout, in RunDotOutputFusion()
|
| D | cpu_compiler.cc | 325 LayoutAssignment::InstructionCanChangeLayout, target_machine_features); in RunHloPassesThroughLayoutAssn()
356 LayoutAssignment::InstructionCanChangeLayout); in RunHloPassesAfterLayoutAssn()
|