Searched refs:replaceAllUsesAfterNodeWith (Results 1 – 15 of 15) sorted by relevance
| /external/pytorch/torch/csrc/jit/passes/ |
| D | remove_inplace_ops.cpp | 71 node->inputs()[0]->replaceAllUsesAfterNodeWith( in RemoveInplaceOps()
|
| D | remove_mutation.cpp | 277 mutated_value->replaceAllUsesAfterNodeWith(node, new_node->output()); in RemoveTensorMutation()
|
| D | vulkan_rewrite.cpp | 152 input->replaceAllUsesAfterNodeWith( in transferInputOutputBackends()
|
| D | specialize_autogradzero.cpp | 67 i->replaceAllUsesAfterNodeWith(opt_pn, opt_pn->output()); in insertProfileNodesForSpecializeAutogradZero()
|
| D | frozen_ops_to_mkldnn.cpp | 744 v->replaceAllUsesAfterNodeWith(from_mkldnn, from_mkldnn->output()); in ComputeSubgraphInMKLDNN()
|
| /external/pytorch/torch/csrc/jit/passes/quantization/ |
| D | insert_quant_dequant.cpp | 182 scalar_value->replaceAllUsesAfterNodeWith(tensor_node, tensor_output); in insertScalarToTensor()
453 original_val->replaceAllUsesAfterNodeWith(dequant, dequant->output()); in insertQuantizationOps()
489 original_val->replaceAllUsesAfterNodeWith(dequant, dequant->output()); in insertQuantizationOps()
1212 original_output->replaceAllUsesAfterNodeWith(quant, quantized_output); in propagateQParams()
1220 output->replaceAllUsesAfterNodeWith(item, scalar); in propagateQParams()
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | scalar_type_analysis.cpp | 374 out->replaceAllUsesAfterNodeWith(cast_node, cast_node->output()); in RecoverScalarTypeForOutput()
|
| D | fixup_onnx_controlflow.cpp | 214 block_output->replaceAllUsesAfterNodeWith(opt_node, opt_node->output()); in ReplaceBlockOutputWithOptional()
|
| D | remove_inplace_ops_for_onnx.cpp | 361 input->replaceAllUsesAfterNodeWith(node, newNode->output()); in PrepareForRemoveMutations()
|
| D | shape_type_inference.cpp | 2452 graph_output->replaceAllUsesAfterNodeWith(opt_node, opt_node->output()); in ReplaceGraphOutputNoneWithOptional()
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | graph_opt.cpp | 469 v->replaceAllUsesAfterNodeWith(deq, deq->output()); in dequantizeResults()
|
| /external/pytorch/torch/csrc/jit/python/ |
| D | python_ir.cpp | 529 .VS(replaceAllUsesAfterNodeWith) in initPythonIRBindings()
|
| /external/pytorch/torch/csrc/jit/ir/ |
| D | ir.h | 282 TORCH_API void replaceAllUsesAfterNodeWith(const Node* node, Value* newValue);
|
| D | ir.cpp | 928 void Value::replaceAllUsesAfterNodeWith(const Node* node, Value* newValue) { in replaceAllUsesAfterNodeWith() function in torch::jit::Value
|
| /external/pytorch/torch/_C/ |
| D | __init__.pyi.in | 764 def replaceAllUsesAfterNodeWith(self, node: Node, val: Value) -> None: ...
|