Home
last modified time | relevance | path

Searched refs:fusion_instr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion.cc192 bool HasOnlyRowMajorLayout(const HloInstruction& fusion_instr) { in HasOnlyRowMajorLayout() argument
193 CHECK(fusion_instr.opcode() == HloOpcode::kFusion); in HasOnlyRowMajorLayout()
194 auto instrs = fusion_instr.fused_instructions_computation()->instructions(); in HasOnlyRowMajorLayout()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc478 auto* fusion_instr = xla::Cast<xla::HloFusionInstruction>(instr); in EmitFusionOp() local
498 *fusion_instr->fused_instructions_computation(), in EmitFusionOp()