Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.h46 HloInstruction* fused_root = fusion->fused_expression_root(); in CanEmitFusedDynamicUpdateSliceInPlace() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc1195 const HloInstruction* fused_root = fusion->fused_expression_root(); in CheckFusionInstruction() local
Dhlo_instructions.cc1135 HloInstruction* fused_root) in HloFusionInstruction()
1475 HloInstruction* fused_root = fused_expression_root(); in CloneAndFuseInternal() local
Dhlo_instruction.cc1201 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3613 const HloInstruction* fused_root = unnested_hlo->fused_expression_root(); in IsUnrollingColumnReductionBeneficial() local