Searched refs:output_fused_op (Results 1 – 1 of 1) sorted by relevance
1793 const HloInstruction* output_fused_op = inst->fused_expression_root(); in BuildGemmThunk() local1797 const HloInstruction* dot = output_fused_op->operand(0); in BuildGemmThunk()1798 if (output_fused_op->opcode() == HloOpcode::kMultiply) { in BuildGemmThunk()1799 const HloInstruction* alpha = output_fused_op->operand(1); in BuildGemmThunk()1818 CHECK_EQ(output_fused_op->opcode(), HloOpcode::kAdd); in BuildGemmThunk()1819 bias = output_fused_op->operand(1); in BuildGemmThunk()