Searched defs:fused_parameter (Results 1 – 4 of 4) sorted by relevance
173 auto fused_parameter = consumer->fused_parameter(producer_operand_index); in ShouldFuseInexpensiveChecks() local
137 auto* fused_parameter = inst->fused_parameter(i); in RevertIfFusionInternalBF16Changes() local248 auto* fused_parameter = in AllUsersConsumeBF16() local
1223 HloInstruction* fused_parameter = in AddFusionOperand() local1340 HloInstruction* HloFusionInstruction::fused_parameter( in fused_parameter() function in xla::HloFusionInstruction1418 HloInstruction* fused_parameter = fused_parameters[operand_num]; in CloneAndFuseInternal() local
3342 HloInstruction* HloInstruction::fused_parameter(int64 parameter_number) const { in fused_parameter() function in xla::HloInstruction