Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc84 HloInstruction* new_lhs = dot->mutable_operand(0); in FoldTransposeIntoDot()
85 HloInstruction* new_rhs = dot->mutable_operand(1); in FoldTransposeIntoDot()
100 new_lhs = new_lhs->mutable_operand(0); in FoldTransposeIntoDot()
106 new_rhs = new_rhs->mutable_operand(0); in FoldTransposeIntoDot()
134 HloInstruction& transpose = *convolution.mutable_operand(kLhsIdx); in FoldTransposeIntoConvolution()
136 HloInstruction& transpose_operand = *transpose.mutable_operand(0); in FoldTransposeIntoConvolution()
151 new_lhs = convolution.mutable_operand(kLhsIdx); in FoldTransposeIntoConvolution()
157 HloInstruction& transpose = *convolution.mutable_operand(kRhsIdx); in FoldTransposeIntoConvolution()
159 HloInstruction& transpose_operand = *transpose.mutable_operand(0); in FoldTransposeIntoConvolution()
174 new_rhs = convolution.mutable_operand(kRhsIdx); in FoldTransposeIntoConvolution()
Ddynamic_index_splitter.cc55 dynamic_op, dynamic_op->mutable_operand(1))); in Run()
58 dynamic_op, dynamic_op->mutable_operand(0))); in Run()
66 auto index_operand = dynamic_op->mutable_operand(index_operand_number); in Run()
85 dynamic_op->shape(), dynamic_op->mutable_operand(0), in Run()
86 dynamic_op->mutable_operand(1), absl::MakeSpan(index_array)) in Run()
88 dynamic_op->shape(), dynamic_op->mutable_operand(0), in Run()
Dalgebraic_simplifier.cc142 instr, operand->mutable_operand(0), options); in BitcastingOperandOfReshapeOrCopyChainHelper()
157 instr, instr->mutable_operand(0), options); in BitcastingOperandOfReshapeOrCopyChain()
467 operand = instruction->mutable_operand(0); in ReplaceWithBitcast()
613 ReplaceInstructionIfSameShape(bitcast, bitcast->mutable_operand(0)); in HandleBitcast()
620 ReplaceInstructionIfSameShape(bitcast, bitcast->mutable_operand(0)); in HandleBitcastConvert()
632 if (ReplaceInstructionIfSameShape(copy, copy->mutable_operand(0))) { in HandleCopy()
685 HloInstruction* slice_operand = operands[i]->mutable_operand(0); in HandleConcatenate()
771 operands[pad_value_operand]->mutable_operand(0), padding_config)); in HandleConcatenate()
1161 lhs = lhs->mutable_operand(0); in HandleDotStrengthReduction()
1173 rhs = rhs->mutable_operand(0); in HandleDotStrengthReduction()
[all …]
Dtuple_simplifier.cc82 HloInstruction* operand = instruction->mutable_operand(operand_number); in Run()
89 top_tuple = operand->mutable_operand(0); in Run()
118 instruction->mutable_operand(0)->mutable_operand( in Run()
Ddynamic_padder.cc56 return inst->mutable_operand(1); in ChooseIdentityValue()
68 return inst->mutable_operand(1); in ChooseIdentityValue()
72 return inst->mutable_operand(2); in ChooseIdentityValue()
118 HloInstruction* operand = inst->mutable_operand(operand_num); in Run()
176 operand = inst->mutable_operand(operand_num); in Run()
Dhlo_query.cc80 *matching_operand = instruction->mutable_operand(0); in MatchBinaryInstructionOperand()
81 *other_operand = instruction->mutable_operand(1); in MatchBinaryInstructionOperand()
85 *matching_operand = instruction->mutable_operand(1); in MatchBinaryInstructionOperand()
86 *other_operand = instruction->mutable_operand(0); in MatchBinaryInstructionOperand()
Dbatchnorm_expander.cc205 HloInstruction* operand = batch_norm->mutable_operand(0); in HandleBatchNormTraining()
210 HloInstruction* scale = batch_norm->mutable_operand(1); in HandleBatchNormTraining()
211 HloInstruction* offset = batch_norm->mutable_operand(2); in HandleBatchNormTraining()
332 HloInstruction* operand = batch_norm->mutable_operand(0); in HandleBatchNormInference()
337 HloInstruction* scale = batch_norm->mutable_operand(1); in HandleBatchNormInference()
338 HloInstruction* offset = batch_norm->mutable_operand(2); in HandleBatchNormInference()
339 HloInstruction* mean = batch_norm->mutable_operand(3); in HandleBatchNormInference()
340 HloInstruction* var = batch_norm->mutable_operand(4); in HandleBatchNormInference()
461 HloInstruction* activation = batch_norm->mutable_operand(0); in HandleBatchNormGrad()
464 HloInstruction* scale = batch_norm->mutable_operand(1); in HandleBatchNormGrad()
[all …]
Dconvolution_group_converter.cc203 auto activation = convolution->mutable_operand(0); in HandleBatchGroupCount()
204 auto filter = convolution->mutable_operand(1); in HandleBatchGroupCount()
335 auto filter = convolution->mutable_operand(1); in HandleConvolution()
381 convolution->shape(), convolution->mutable_operand(0), new_filter, in HandleConvolution()
402 auto filter = convolution->mutable_operand(1); in HandleConvolution()
403 auto activation = convolution->mutable_operand(0); in HandleConvolution()
475 auto activation = convolution->mutable_operand(0); in HandleConvolution()
Dar_crs_combiner.cc137 GetAllTuples(tuple->mutable_operand(instruction->tuple_index())); in GetAllTuples()
158 if (!InstructionsComputeSameValue(tuple->mutable_operand(i1), in TupleElementsComputeSameValue()
159 tuple->mutable_operand(i2), in TupleElementsComputeSameValue()
327 auto prev = all_reduce->mutable_operand(0); in RewriteGraph()
350 other_operand->mutable_operand(0))); in RewriteGraph()
Dinstruction_fusion.cc212 auto* consumer_operand = consumer->mutable_operand(i); in CanFuseOnAllPaths()
406 if (!ContainsKey(post_order_index_, instruction->mutable_operand(i))) { in DequeueNextInstructionAndOperandsToFuseInOrder()
415 FindOrDie(post_order_index_, instruction->mutable_operand(i)) > in DequeueNextInstructionAndOperandsToFuseInOrder()
416 FindOrDie(post_order_index_, instruction->mutable_operand(j))); in DequeueNextInstructionAndOperandsToFuseInOrder()
488 HloInstruction* operand = instruction->mutable_operand(i); in Run()
612 HloInstruction* producer = consumer->mutable_operand(operand_index); in ShouldFuse()
Dhlo_domain_remover.cc82 operand->ReplaceAllUsesWith(operand->mutable_operand(0))); in Run()
91 computation->set_root_instruction(root->mutable_operand(0)); in Run()
Dbatch_dot_simplification.cc27 HloInstruction *lhs = batch_dot->mutable_operand(0), in ElideDegenerateBatchDimensionFromBatchDot()
28 *rhs = batch_dot->mutable_operand(1); in ElideDegenerateBatchDimensionFromBatchDot()
Ddot_decomposer.cc37 HloInstruction* lhs = dot->mutable_operand(0); in DecomposeBatchDot()
38 HloInstruction* rhs = dot->mutable_operand(1); in DecomposeBatchDot()
210 original_dot->mutable_operand(0), lhs_transpose)); in CanonicalizeDot()
254 original_dot->mutable_operand(1), rhs_transpose)); in CanonicalizeDot()
Dwhile_loop_simplifier.cc56 HloInstruction* while_init = while_op->mutable_operand(0); in TryRemoveDeadWhileParams()
241 while_body_root->mutable_operand(old_idx)); in TryRemoveDeadWhileParams()
313 auto* while_init = while_op->mutable_operand(0); in TryRemoveConstantParams()
384 tuple_elems.push_back(while_init->mutable_operand(i)); in TryRemoveConstantParams()
487 while_op, while_op->mutable_operand(0))); in TryRemoveWhileLoop()
645 auto* while_init = while_op->mutable_operand(0); in TryFlattenNestedTuples()
772 auto* while_init = while_op->mutable_operand(0); in TryMergeInductionVariables()
793 if (!Match(while_body_root->mutable_operand(i), in TryMergeInductionVariables()
Dinstruction_fusion_test.cc59 HloInstruction* add = sub->mutable_operand(0); in TEST_F()
83 HloInstruction* abs = root->mutable_operand(0); in TEST_F()
103 HloInstruction* abs = root->mutable_operand(0); in TEST_F()
104 HloInstruction* tanh = root->mutable_operand(1); in TEST_F()
Dbfloat16_conversion_folding.cc82 auto operand = hlo->mutable_operand(operand_index); in FoldOperandConversion()
85 hlo->ReplaceOperandWith(operand_index, operand->mutable_operand(0))); in FoldOperandConversion()
Dscatter_expander.cc347 HloInstruction* operand = scatter->mutable_operand(0); in ExpandScatter()
348 HloInstruction* scatter_indices = scatter->mutable_operand(1); in ExpandScatter()
349 HloInstruction* updates = scatter->mutable_operand(2); in ExpandScatter()
Dwhile_loop_invariant_code_motion.cc54 HloInstruction* while_operand = while_instr->mutable_operand(0); in CreateLoopInvariantCopy()
91 frame->instruction->mutable_operand(frame->operand_index++); in CreateLoopInvariantCopy()
Dhlo_domain_isolator.cc44 root = root->mutable_operand(0); in RunInternal()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter.cc272 HloInstruction* reverse_filter = conv->mutable_operand(1); in MatchBackwardInput()
458 rhs = rhs->mutable_operand(0); in MatchBackwardInput()
510 conv->mutable_operand(0), conv->mutable_operand(1), in RunOnInstruction()
518 conv->mutable_operand(0), rhs, window, dnums, in RunOnInstruction()
525 conv->mutable_operand(0), conv->mutable_operand(1), in RunOnInstruction()
Dcudnn_conv_pad_for_tensor_cores.cc81 auto* lhs = conv->mutable_operand(0); in PadConv()
82 auto* rhs = conv->mutable_operand(1); in PadConv()
124 auto* lhs = conv->mutable_operand(0); in PadForTensorCores()
125 auto* rhs = conv->mutable_operand(1); in PadForTensorCores()
Dcudnn_conv_padding_legalization.cc160 conv->mutable_operand(0)); in CanonicalizeForwardConvolution()
163 conv->mutable_operand(1)); in CanonicalizeForwardConvolution()
218 HloInstruction* input = backward_conv->mutable_operand(0); in CanonicalizeBackwardFilterConvolution()
252 HloInstruction* output = backward_conv->mutable_operand(1); in CanonicalizeBackwardFilterConvolution()
324 HloInstruction* output = backward_conv->mutable_operand(0); in CanonicalizeBackwardInputConvolution()
325 HloInstruction* filter = backward_conv->mutable_operand(1); in CanonicalizeBackwardInputConvolution()
Dcudnn_fused_conv_rewriter.cc219 std::vector<HloInstruction*> args = {conv->mutable_operand(0), in TryRewriteToCudnnForwardRelu()
220 conv->mutable_operand(1), bias}; in TryRewriteToCudnnForwardRelu()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_hlo_test.cc81 module->entry_computation()->root_instruction()->mutable_operand(0); in XLA_TEST_P()
91 reduce_instruction->mutable_operand(0)->mutable_shape(); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc57 HloInstruction* input = hlo->mutable_operand(0); in Run()
78 HloInstruction* kernel = hlo->mutable_operand(1); in Run()

123