Home
last modified time | relevance | path

Searched refs:AppendOperand (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc79 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 …]
Dhlo_instruction.cc595 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 …]
Dhlo_instruction.h1631 void AppendOperand(HloInstruction* operand);