Home
last modified time | relevance | path

Searched refs:mutable_operand (Results 1 – 25 of 121) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/xla/service/
Dfusion_node_indexing_evaluation_test.cc90 HloInstruction* add = sub->mutable_operand(0); in TEST_F()
106 HloInstruction* slice1 = sub->mutable_operand(0); in TEST_F()
107 HloInstruction* slice2 = sub->mutable_operand(1); in TEST_F()
147 HloInstruction* slice2_0 = add2->mutable_operand(0); in TEST_F()
148 HloInstruction* slice2_1 = add2->mutable_operand(1); in TEST_F()
156 HloInstruction* add1 = fusion->mutable_operand(0); in TEST_F()
161 HloInstruction* slice1_0 = fusion->mutable_operand(0); in TEST_F()
167 HloInstruction* slice1_1 = fusion->mutable_operand(0); in TEST_F()
173 HloInstruction* add0 = fusion->mutable_operand(0); in TEST_F()
204 HloInstruction* add0 = fusion->mutable_operand(0); in TEST_F()
Ddynamic_index_splitter.cc57 dynamic_op, dynamic_op->mutable_operand(1))); in Run()
60 dynamic_op, dynamic_op->mutable_operand(0))); in Run()
69 auto index_operand = dynamic_op->mutable_operand(index_operand_number); in Run()
89 dynamic_op->shape(), dynamic_op->mutable_operand(0), in Run()
90 dynamic_op->mutable_operand(1), absl::MakeSpan(index_array)) in Run()
92 dynamic_op->shape(), dynamic_op->mutable_operand(0), in Run()
Dtranspose_folding.cc86 HloInstruction* lhs = dot->mutable_operand(0); in FoldTransposeIntoDot()
87 HloInstruction* rhs = dot->mutable_operand(1); in FoldTransposeIntoDot()
95 lhs = lhs->mutable_operand(0); in FoldTransposeIntoDot()
102 rhs = rhs->mutable_operand(0); in FoldTransposeIntoDot()
130 HloInstruction& transpose = *convolution.mutable_operand(kLhsIdx); in FoldTransposeIntoConvolution()
132 HloInstruction& transpose_operand = *transpose.mutable_operand(0); in FoldTransposeIntoConvolution()
147 new_lhs = convolution.mutable_operand(kLhsIdx); in FoldTransposeIntoConvolution()
153 HloInstruction& transpose = *convolution.mutable_operand(kRhsIdx); in FoldTransposeIntoConvolution()
155 HloInstruction& transpose_operand = *transpose.mutable_operand(0); in FoldTransposeIntoConvolution()
170 new_rhs = convolution.mutable_operand(kRhsIdx); in FoldTransposeIntoConvolution()
Ddynamic_dimension_simplifier.cc58 auto concat = slice->mutable_operand(0); in SliceConcatForwarding()
97 auto broadcast = reshape->mutable_operand(0); in ReshapeBroadcastForwarding()
110 if (broadcast->mutable_operand(0)->shape().rank() != 0) { in ReshapeBroadcastForwarding()
115 reshape->ReplaceAllUsesWith(broadcast->mutable_operand(0))); in ReshapeBroadcastForwarding()
125 auto reshape_2 = reshape->mutable_operand(0); in ReshapeReshapeForwarding()
134 reshape->ReplaceAllUsesWith(reshape_2->mutable_operand(0))); in ReshapeReshapeForwarding()
144 auto operand = convert->mutable_operand(0); in IdentityConvertRemoving()
157 auto operand = reshape->mutable_operand(0); in IdentityReshapeRemoving()
Dalgebraic_simplifier.cc281 instr, operand->mutable_operand(0), options); in BitcastingOperandOfReshapeOrCopyChainHelper()
296 instr, instr->mutable_operand(0), options); in BitcastingOperandOfReshapeOrCopyChain()
479 auto lhs = dot->mutable_operand(0); in ScalarMultiplyReduction()
480 auto rhs = dot->mutable_operand(1); in ScalarMultiplyReduction()
626 operand = instruction->mutable_operand(0); in ReplaceWithBitcast()
674 HloInstruction* abs_operand = abs->mutable_operand(0); in HandleAbs()
763 sliced_lhs, rhs->mutable_operand(1))); in HandleAdd()
1049 TF_RETURN_IF_ERROR(HandleBitcast(bitcast->mutable_operand(0))); in HandleBitcast()
1057 ReplaceInstructionIfCompatible(bitcast, bitcast->mutable_operand(0)); in HandleBitcast()
1249 HloInstruction* bitcast = root_copy->mutable_operand(0); in SwapCopyBitcastCopy()
[all …]
Dall_reduce_reassociate.cc67 auto *ar0 = Cast<HloAllReduceInstruction>(inst->mutable_operand(0)); in Run()
68 auto *ar1 = Cast<HloAllReduceInstruction>(inst->mutable_operand(1)); in Run()
81 inst->CloneWithNewOperands(inst->shape(), {ar0->mutable_operand(0), in Run()
82 ar1->mutable_operand(0)})); in Run()
Dspace_to_batch_converter.cc478 auto activations = convolution->mutable_operand(0); in IsThisBackPropFilterConv()
479 auto kernel = convolution->mutable_operand(1); in IsThisBackPropFilterConv()
909 if (old_to_new_instrs_.contains(instr->mutable_operand(0))) { in Run()
910 producer = instr->mutable_operand(0); in Run()
911 } else if (old_to_new_instrs_.contains(instr->mutable_operand(1))) { in Run()
912 producer = instr->mutable_operand(1); in Run()
928 if (old_to_new_instrs_.count(instr->mutable_operand(i))) { in Run()
930 BatchToSpace(instr->mutable_operand(i))); in Run()
959 auto old_producer = consumer->mutable_operand(i); in CanPropagate()
1036 if (!instr_to_dim_map_.contains(consumer->mutable_operand(i))) { in CanPropagate()
[all …]
Dwhile_loop_all_reduce_code_motion_test.cc105 transformed_while->mutable_operand(0)->mutable_operand(3); in TEST_F()
116 moved_all_reduce->mutable_operand(0)) in TEST_F()
399 transformed_while->mutable_operand(0)->mutable_operand(3); in TEST_F()
489 transformed_while->mutable_operand(0)->mutable_operand(3); in TEST_F()
660 transformed_while->mutable_operand(0)->mutable_operand(3); in TEST_F()
741 transformed_while->mutable_operand(0)->mutable_operand(3); in TEST_F()
822 transformed_while->mutable_operand(0)->mutable_operand(3); in TEST_F()
Dlayout_normalization.cc150 auto operand = hlo->mutable_operand(0); in HandleBroadcast()
187 auto operand = hlo->mutable_operand(0); in HandleElementwiseUnary()
231 auto a = hlo->mutable_operand(0); in HandleElementwiseBinary()
232 auto b = hlo->mutable_operand(1); in HandleElementwiseBinary()
260 auto operand = hlo->mutable_operand(0); in HandleReshape()
297 auto operand = hlo->mutable_operand(0); in HandleTranspose()
340 auto operand = hlo->mutable_operand(0); in HandleCopy()
362 auto operand = hlo->mutable_operand(0); in HandleReverse()
385 auto operand = hlo->mutable_operand(0); in HandlePad()
387 auto padded_by = hlo->mutable_operand(1); in HandlePad()
[all …]
Ddynamic_padder.cc76 return inst->mutable_operand(2); in ChooseIdentityValue()
101 return inst->mutable_operand(init_value_index); in ChooseIdentityValue()
110 return inst->mutable_operand(init_value_index); in ChooseIdentityValue()
126 return inst->mutable_operand(1); in ChooseIdentityValue()
180 HloInstruction* operand = instr->mutable_operand(0); in ReplaceGetSize()
211 HloInstruction* operand = instr->mutable_operand(0); in ReplaceSetSize()
227 HloInstruction* operand = instr->mutable_operand(0); in ReplaceSetBound()
553 operand_shape, reshape->mutable_operand(0), operand_static_dim_size, in RewriteDynamicReshapeSplitInput()
963 HloInstruction* grad = custom_call_conv->mutable_operand(1); in RewriteDynamicConvolutionInputGrad()
964 HloInstruction* kernel = custom_call_conv->mutable_operand(2); in RewriteDynamicConvolutionInputGrad()
[all …]
Dall_reduce_simplifier.cc65 computation->ReplaceInstruction(inst, inst->mutable_operand(0))); in Run()
98 all_reduce, all_reduce->mutable_operand(0))); in Run()
129 all_reduce->mutable_operand(0), multiplier)); in Run()
136 replacement = all_reduce->mutable_operand(0); in Run()
Dbatchnorm_expander.cc171 HloInstruction* operand = batch_norm->mutable_operand(0); in HandleBatchNormTraining()
176 HloInstruction* scale = batch_norm->mutable_operand(1); in HandleBatchNormTraining()
177 HloInstruction* offset = batch_norm->mutable_operand(2); in HandleBatchNormTraining()
300 HloInstruction* operand = batch_norm->mutable_operand(0); in HandleBatchNormInference()
305 HloInstruction* scale = batch_norm->mutable_operand(1); in HandleBatchNormInference()
306 HloInstruction* offset = batch_norm->mutable_operand(2); in HandleBatchNormInference()
307 HloInstruction* mean = batch_norm->mutable_operand(3); in HandleBatchNormInference()
308 HloInstruction* var = batch_norm->mutable_operand(4); in HandleBatchNormInference()
413 HloInstruction* activation = batch_norm->mutable_operand(0); in HandleBatchNormGrad()
416 HloInstruction* scale = batch_norm->mutable_operand(1); in HandleBatchNormGrad()
[all …]
Dtuple_simplifier.cc39 HloInstruction* operand = tuple->mutable_operand(operand_number); in RemoveWholeTuple()
45 top_tuple = operand->mutable_operand(0); in RemoveWholeTuple()
105 replacement = ancestor.first->mutable_operand(ancestor.second[0]); in Run()
Dhlo_query.cc87 *matching_operand = instruction->mutable_operand(0); in MatchBinaryInstructionOperand()
88 *other_operand = instruction->mutable_operand(1); in MatchBinaryInstructionOperand()
92 *matching_operand = instruction->mutable_operand(1); in MatchBinaryInstructionOperand()
93 *other_operand = instruction->mutable_operand(0); in MatchBinaryInstructionOperand()
Dasync_collective_creator.cc115 HloInstruction* operand = cp->mutable_operand(0); in Run()
136 operand, cp->mutable_operand(1), cp->mutable_operand(2), in Run()
137 cp->mutable_operand(3), cp->source_target_pairs(), in Run()
162 HloInstruction* async_start = async_done->mutable_operand(0); in Run()
Dconditional_to_select.cc47 conditional->shape(), {conditional->mutable_operand(1)}, in DoConditionalToSelect()
52 conditional->shape(), {conditional->mutable_operand(2)}, in DoConditionalToSelect()
55 HloInstruction* condition = conditional->mutable_operand(0); in DoConditionalToSelect()
Dconvert_operand_folding.cc52 auto* operand = instruction->mutable_operand(i); in ExpandInstruction()
55 i, operand->mutable_operand(0))); in ExpandInstruction()
Dhlo_schedule_test.cc322 module->entry_computation()->root_instruction()->mutable_operand(0); in TEST_F()
323 HloInstruction* init = xla_while->mutable_operand(0); in TEST_F()
397 ->mutable_operand(0) in TEST_F()
398 ->mutable_operand(0); in TEST_F()
399 HloInstruction* init = xla_while->mutable_operand(0); in TEST_F()
Dwhile_loop_all_reduce_code_motion.cc232 instruction = instruction->mutable_operand(0); in IsAllReduceMovable()
242 instruction = instruction->mutable_operand(0); in IsAllReduceMovable()
255 instruction = instruction->mutable_operand(0); in IsAllReduceMovable()
503 user->mutable_operand(buffer_index); in IsAllReduceMovable()
567 HloInstruction* old_while_init = old_while_instruction->mutable_operand(0); in CreateNewWhileInit()
578 HloInstruction* old_buffer = old_while_init->mutable_operand(tuple_index); in CreateNewWhileInit()
589 new_while_init_elements[i] = old_while_init->mutable_operand(i); in CreateNewWhileInit()
591 tuple_index_to_old_buffer[i] = old_while_init->mutable_operand(i); in CreateNewWhileInit()
831 all_reduce, all_reduce->mutable_operand(0))); in Run()
Dlogistic_expander.cc42 HloInstruction* operand = logistic->mutable_operand(0); in ExpandLogisticWithTanh()
58 HloInstruction* operand = logistic->mutable_operand(0); in ExpandLogisticWithExp()
Dwhile_loop_simplifier.cc62 HloInstruction* while_init = while_op->mutable_operand(0); in RemoveDeadTupleIndices()
149 while_body_root->mutable_operand(old_idx)); in RemoveDeadTupleIndices()
237 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveDeadWhileParams()
378 while_body_root->mutable_operand(inst->tuple_index()); in TryRemoveDeadWhileParams()
398 HloInstruction* output = while_body_root->mutable_operand(i); in TryRemoveDeadWhileParams()
418 HloInstruction* output = while_body_root->mutable_operand(i); in TryRemoveDeadWhileParams()
476 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveRepeatedWhileTupleIndicesHelper()
535 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveRepeatedWhileTupleIndices()
615 while_init = while_op->mutable_operand(0); in TryRemoveRepeatedWhileTupleIndices()
631 auto* while_init = while_op->mutable_operand(0); in TryRemoveConstantParams()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreduction_splitter.cc46 HloInstruction *operand = reduce->mutable_operand(0); in HandleReduce()
87 pre_reduce_shape, reduce->mutable_operand(0), in HandleReduce()
88 reduce->mutable_operand(1), pre_reduce_dims, reduce->to_apply()); in HandleReduce()
105 reduce->mutable_operand(1), final_reduce_dims, reduce->to_apply()); in HandleReduce()
Dgpu_conv_rewriter.cc250 HloInstruction* lhs = conv->mutable_operand(0); in MatchBackwardFilter()
275 HloInstruction* reverse_filter = conv->mutable_operand(1); in MatchBackwardInput()
488 rhs = rhs->mutable_operand(0); in MatchBackwardInput()
609 auto lhs = conv->mutable_operand(0); in ConvertBatchGroupedToFeatureGroupedConvolution()
610 auto rhs = conv->mutable_operand(1); in ConvertBatchGroupedToFeatureGroupedConvolution()
675 conv->mutable_operand(0), rhs, window, dnums, in CreateCustomCallHelper()
683 conv->mutable_operand(1), window, dnums, in CreateCustomCallHelper()
694 conv->mutable_operand(0), conv->mutable_operand(1), in CreateCustomCallHelper()
Dgpu_conv_padding_legalization.cc163 conv->mutable_operand(0)); in CanonicalizeForwardConvolution()
166 conv->mutable_operand(1)); in CanonicalizeForwardConvolution()
221 HloInstruction* input = backward_conv->mutable_operand(0); in CanonicalizeBackwardFilterConvolution()
255 HloInstruction* output = backward_conv->mutable_operand(1); in CanonicalizeBackwardFilterConvolution()
331 HloInstruction* output = backward_conv->mutable_operand(0); in CanonicalizeBackwardInputConvolution()
332 HloInstruction* filter = backward_conv->mutable_operand(1); in CanonicalizeBackwardInputConvolution()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_hlo_test.cc78 module->entry_computation()->root_instruction()->mutable_operand(0); in XLA_TEST_P()
88 reduce_instruction->mutable_operand(0)->mutable_shape(); in XLA_TEST_P()

12345