Home
last modified time | relevance | path

Searched refs:CloneWithNewOperands (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_element_type_converter.cc190 hlo->CloneWithNewOperands(shape, new_operands, &context)); in Run()
200 hlo->CloneWithNewOperands(new_shape, new_operands, &context)); in Run()
208 hlo->CloneWithNewOperands(hlo->shape(), new_operands, &context)); in Run()
Dreshape_mover.cc164 operand->CloneWithNewOperands(new_shape, operand->operands())); in UpdateOperand()
174 operand->CloneWithNewOperands(new_shape, operand->operands())); in UpdateOperand()
226 computation->AddInstruction(instruction->CloneWithNewOperands( in PerformSinkReshapeOrTranspose()
Dstable_sort_expander.cc107 sort->CloneWithNewOperands(new_sort_shape, new_operands)); in ExpandInstruction()
173 inst->CloneWithNewOperands(inst->shape(), new_operands); in ExpandInstruction()
Ddefuser.cc76 fused_instruction->CloneWithNewOperands(fused_instruction->shape(), in Defuse()
Dmap_inliner.cc103 root.CloneWithNewOperands(map->shape(), params)); in HandleMap()
Dcall_inliner.cc51 auto new_hlo = hlo->CloneWithNewOperands(hlo->shape(), new_operands); in DefaultAction()
Dsort_simplifier.cc82 sort->CloneWithNewOperands(new_sort_shape, operands)); in RemoveUnusedOperandFromSort()
Dwhile_loop_invariant_code_motion.cc75 parent_of_while->AddInstruction(old_instruction->CloneWithNewOperands( in CreateLoopInvariantCopy()
Dalgebraic_simplifier.cc667 computation_->AddInstruction(concatenate->CloneWithNewOperands( in HandleConcatenate()
733 computation_->AddInstruction(concatenate->CloneWithNewOperands( in HandleConcatenate()
2442 computation_->AddInstruction(user->CloneWithNewOperands( in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
3452 auto new_conv = convolution->CloneWithNewOperands( in FoldConvInputPad()
3518 auto new_conv = convolution->CloneWithNewOperands( in FoldConvFilterPad()
3703 auto clone = map_root->CloneWithNewOperands(map_root->shape(), {}); in HandleMap()
3725 auto clone = map_root->CloneWithNewOperands(map->shape(), new_operands); in HandleMap()
Dhlo_computation.cc1013 instr->CloneWithNewOperands(instr->shape(), new_operands, context)); in CloneWithReplacements()
Dhlo_instruction.h1198 std::unique_ptr<HloInstruction> CloneWithNewOperands(
Dhlo_instruction.cc1333 std::unique_ptr<HloInstruction> HloInstruction::CloneWithNewOperands( in CloneWithNewOperands() function in xla::HloInstruction
1541 CloneWithNewOperands(shape_, operands_, context); in Clone()
Dhlo_evaluator.cc330 instruction->CloneWithNewOperands(instruction->shape(), operands); in EvaluateWithSubstitutions()
Dhlo_instruction_test.cc707 auto cloned = fusion->CloneWithNewOperands(fusion->shape(), {}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_padding_legalization.cc184 conv->CloneWithNewOperands(conv->shape(), operands)); in CanonicalizeForwardConvolution()
262 computation->AddInstruction(backward_conv->CloneWithNewOperands( in CanonicalizeBackwardFilterConvolution()
328 computation->AddInstruction(backward_conv->CloneWithNewOperands( in CanonicalizeBackwardInputConvolution()
Dvariadic_op_splitter.cc68 auto new_concat = comp->AddInstruction(concat->CloneWithNewOperands( in SplitConcatenate()
Dcudnn_conv_pad_for_tensor_cores.cc96 add(conv->CloneWithNewOperands(new_conv_shape, {new_lhs, new_rhs})); in PadConv()
Dcudnn_conv_algorithm_picker.cc389 instr->CloneWithNewOperands(new_call_shape, instr->operands())); in RunOnInstruction()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc75 hlo->CloneWithNewOperands(hlo->shape(), new_operands, &clone_context_); in DefaultAction()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcustom_call_test.cc178 custom_call->CloneWithNewOperands(r2f32_dim0_major, {custom_call})); in XLA_TEST_F()