Searched defs:new_operand (Results 1 – 10 of 10) sorted by relevance
107 auto new_operand = in HandleConcatenate() local400 auto new_operand = operand_s_normalized == a0->shape() in HandlePad() local
307 auto new_operand = in CopyIntoConditional() local322 auto new_operand = root->mutable_operand(index); in CopyIntoConditional() local
1976 for (const HloInstruction* new_operand : new_operands) { in CloneWithNewOperands() local2688 HloInstruction* new_operand) { in ReplaceOperandWith()2698 int64_t operand_num, HloInstruction* new_operand) { in ReplaceOperandWithDifferentShape()4555 HloInstruction* HloInstruction::AddFusionOperand(HloInstruction* new_operand) { in AddFusionOperand()
2794 HloInstruction* new_operand = gather->mutable_operand(0); in HandleGather() local3391 HloInstruction* new_operand = in HandleOptimizationBarrier() local3560 auto new_operand = in HandleBroadcast() local4355 auto new_operand = reshape->AddInstruction(HloInstruction::CreateReshape( in HandleReshape() local
1596 HloInstruction* new_operand) { in AddCallOperand()1915 HloInstruction* new_operand) { in AddFusionOperand()2016 auto new_operand = in MergeFusionInstructionIntoMultiOutput() local
357 auto new_operand = old_to_new_instrs_[old_operand]; in GetSpatialDimsToSplit() local1691 auto new_operand = old_to_new_instrs_[first_operand]; in SupportedOpForPropagation() local
1255 const HloInstruction* new_operand = replace(operand); in CloneWithReplacements() local
1484 HloInstruction* new_operand = hlo->mutable_operand(operand_index); in HandleConditional() local
136 auto new_operand = ConvertCoreRTTensorHandleToFallbackTensor( in ConvertFallbackOperands() local
4071 for (auto* new_operand : find_result.second) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local