| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | frozen_conv_folding.cpp | 128 conv->replaceInputWith(conv_w_value, fused_conv_w); in FoldFrozenConvBatchnorm()
129 conv->replaceInputWith(conv_b_value, fused_conv_b); in FoldFrozenConvBatchnorm()
272 add_or_sub->replaceInputWith( in FoldFrozenConvAddOrSub()
287 conv->replaceInputWith(conv_b_value, fused_conv_b); in FoldFrozenConvAddOrSub()
343 mul_or_div->replaceInputWith( in FoldFrozenConvMulOrDiv()
357 conv->replaceInputWith(conv_weight_value, fused_conv_weight); in FoldFrozenConvMulOrDiv()
381 conv->replaceInputWith(conv_b_value, fused_conv_bias); in FoldFrozenConvMulOrDiv()
|
| H A D | frozen_linear_folding.cpp | 124 linear->replaceInputWith(linear_w_value, fused_linear_w); in FoldFrozenLinearBatchnorm()
125 linear->replaceInputWith(linear_b_value, fused_linear_b); in FoldFrozenLinearBatchnorm()
|
| H A D | autocast.cpp | 174 node->replaceInputWith(input, new_input); in castTensorInputs()
183 node->replaceInputWith(input, new_input); in castTensorInputs()
|
| H A D | integer_value_refinement.cpp | 147 n->replaceInputWith(input_v, refine_constant); in RefineIntegerValues()
|
| H A D | bailout_graph.cpp | 180 adj_iter_ctr->node()->replaceInputWith(one, new_lv.currentTripCount()); in buildBailOutLoop()
|
| H A D | frozen_ops_to_mkldnn.cpp | 629 n->replaceInputWith(input, mkldnn_tensor_value); in replaceInputWithMKLDNNTensor()
642 n->replaceInputWith(input, mkldnn_tensor_value); in replaceInputWithMKLDNNTensor()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | fixup_onnx_controlflow.cpp | 38 consumer_node->replaceInputWith(cond_val, cast_node->output()); in InsertCastForCond()
141 sub_block->return_node()->replaceInputWith( in ConvertSequenceDependencies()
241 block->return_node()->replaceInputWith(output, id_node->output()); in FixupONNXSubblockOutputs()
344 node->replaceInputWith(input, opt_node->output()); in FixupONNXLoopNodeInputs()
|
| H A D | scalar_type_analysis.cpp | 337 n->replaceInputWith(input, const_node->output()); in UpdateScalarTypeForInputs()
346 n->replaceInputWith(input, cast_node->output()); in UpdateScalarTypeForInputs()
|
| H A D | peephole.cpp | 929 origLogSoftmaxNode->replaceInputWith( in fuseLogSoftmaxNllLoss()
1015 b->return_node()->replaceInputWith(out, id_node->output()); in insertIdentityForInputUsedAsOutput()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
| H A D | defer_size_check.cpp | 57 node->replaceInputWith(input, shapePreserveOp->output(0)); in analyzeNode()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | autodiff.cpp | 537 node->replaceInputWith(input, lifted_constant->output()); in liftConstants()
583 node->replaceInputWith(input, size); in foldSizeIfNotEqual()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_ir.cpp | 664 .NS(replaceInputWith) in initPythonIRBindings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | ir.h | 594 void replaceInputWith(Value* from, Value* to);
|
| H A D | ir.cpp | 1431 void Node::replaceInputWith(Value* from, Value* to) { in replaceInputWith() function in torch::jit::Node
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | insert_quant_dequant.cpp | 539 user->replaceInputWith(dequant_out, std::get<2>(quant_ops)->output()); in ReplicateChooseQParamsQuantDequant()
|
| /aosp_15_r20/external/pytorch/torch/_C/ |
| H A D | __init__.pyi.in | 803 def replaceInputWith(self, from_: Value, to: Value) -> None: ...
|