Searched refs:AppendOperand (Results 1 – 5 of 5) sorted by relevance
82 AppendOperand(operand); in HloBatchNormInstruction()83 AppendOperand(scale); in HloBatchNormInstruction()113 AppendOperand(offset); in HloBatchNormTrainingInstruction()132 AppendOperand(offset); in HloBatchNormInferenceInstruction()133 AppendOperand(mean); in HloBatchNormInferenceInstruction()134 AppendOperand(variance); in HloBatchNormInferenceInstruction()153 AppendOperand(mean); in HloBatchNormGradInstruction()154 AppendOperand(variance); in HloBatchNormGradInstruction()155 AppendOperand(grad_output); in HloBatchNormGradInstruction()173 AppendOperand(operand); in HloFftInstruction()[all …]
769 instruction->AppendOperand(instruction_map.at(operand_id)); in CreateFromProto()878 instruction->AppendOperand(operand); in CreateNary()1166 instruction->AppendOperand(operand); in CreateAfterAll()1181 instruction->AppendOperand(data_operand); in CreateAddDependency()1182 instruction->AppendOperand(token_operand); in CreateAddDependency()1191 instruction->AppendOperand(init); in CreateWhile()1204 instruction->AppendOperand(pred); in CreateConditional()1205 instruction->AppendOperand(true_computation_arg); in CreateConditional()1206 instruction->AppendOperand(false_computation_arg); in CreateConditional()1221 instruction->AppendOperand(branch_index); in CreateConditional()[all …]
1913 void AppendOperand(HloInstruction* operand);
753 new_root->AppendOperand(new_op); in MoveInstructionIn()
1221 sort->AppendOperand(lt); in RewriteDynamicSort()