Searched refs:AppendOperand (Results 1 – 3 of 3) sorted by relevance
79 AppendOperand(operand); in HloBatchNormInstruction()80 AppendOperand(scale); in HloBatchNormInstruction()110 AppendOperand(offset); in HloBatchNormTrainingInstruction()129 AppendOperand(offset); in HloBatchNormInferenceInstruction()130 AppendOperand(mean); in HloBatchNormInferenceInstruction()131 AppendOperand(variance); in HloBatchNormInferenceInstruction()150 AppendOperand(mean); in HloBatchNormGradInstruction()151 AppendOperand(variance); in HloBatchNormGradInstruction()152 AppendOperand(grad_output); in HloBatchNormGradInstruction()170 AppendOperand(operand); in HloFftInstruction()[all …]
595 instruction->AppendOperand(instruction_map.at(operand_id)); in CreateFromProto()680 instruction->AppendOperand(operand); in CreateNary()920 instruction->AppendOperand(operand); in CreateAfterAll()935 instruction->AppendOperand(data_operand); in CreateAddDependency()936 instruction->AppendOperand(token_operand); in CreateAddDependency()945 instruction->AppendOperand(init); in CreateWhile()958 instruction->AppendOperand(pred); in CreateConditional()959 instruction->AppendOperand(true_computation_arg); in CreateConditional()960 instruction->AppendOperand(false_computation_arg); in CreateConditional()975 instruction->AppendOperand(branch_index); in CreateConditional()[all …]
1631 void AppendOperand(HloInstruction* operand);