Searched defs:fused_expression_root (Results 1 – 3 of 3) sorted by relevance
96 auto fused_expression_root = instr->fused_expression_root(); in ShapesCompatibleForMultiOutputFusion() local
3322 HloInstruction* HloInstruction::fused_expression_root() const { in fused_expression_root() function in xla::HloInstruction
1336 HloInstruction* HloFusionInstruction::fused_expression_root() const { in fused_expression_root() function in xla::HloFusionInstruction