| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | decompose_ops.cpp | 112 insertGraph(*it->owningGraph(), *d_graph, it->inputs()).at(0); in DecomposeOps()
141 Value* new_output = insertGraph(*graph, *d_graph, inputs).at(0); in DecomposeOps()
176 Value* new_output = insertGraph(*graph, *d_graph, inputs).at(0); in DecomposeOps()
|
| H A D | inline_fork_wait.cpp | 21 auto output = insertGraph(*graph, *subgraph, node->inputs()); in InlineForkWait()
|
| H A D | replacement_of_old_operators.cpp | 71 auto new_outputs = insertGraph( in run()
|
| H A D | symbolic_shape_runtime_fusion.cpp | 54 auto sym_shape_values = insertGraph( in InsertSymbolicShapesCompute()
400 const auto subgraph_outputs = insertGraph( in insertDynamicShapesGuard()
|
| H A D | subgraph_rewrite.cpp | 173 insertGraph(*graph, replacement_graph, inputs); in rewriteSinglePatternOnGraph()
|
| H A D | fixup_trace_scope_blocks.cpp | 459 auto new_outputs = insertGraph(*n->owningGraph(), *graph, n->inputs()); in inlineScopeBlocks()
|
| H A D | symbolic_shape_analysis.cpp | 1038 insertGraph( in joinPartialEvaluatedShapeGraphToLargeShapeGraph()
|
| H A D | tensorexpr_fuser.cpp | 372 const auto subgraph_outputs = insertGraph( in insertTypeGuard()
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_python_ir.py | 95 outputs = g.insertGraph(unrolled_mul.graph, list(mul.inputs()))
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | autodiff.cpp | 148 new_outputs = insertGraph(*graph, *fw_graph, node->inputs()); in build_script_grad()
167 auto grad_inputs = insertGraph(*graph, *bw_graph, grad); in build_script_grad()
|
| H A D | decomposition_registry.cpp | 83 insertGraph(*n->owningGraph(), *decomposition->get(), n->inputs()); in DecomposeOp()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | ir.cpp | 2094 std::vector<torch::jit::Value*> new_outputs = insertGraph( in inlineCallTo()
2216 std::vector<Value*> insertGraph( in insertGraph() function
2241 std::vector<Value*> insertGraph( in insertGraph() function
2246 return insertGraph(g, callee, inputs, value_map); in insertGraph()
|
| H A D | ir.h | 1606 TORCH_API std::vector<Value*> insertGraph(
1610 TORCH_API std::vector<Value*> insertGraph(
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_ir.cpp | 463 return insertGraph(g, callee, inputs); in initPythonIRBindings()
471 return insertGraph(g, callee, inputs, value_map); in initPythonIRBindings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | schema_matching.cpp | 771 return insertGraph( in emitBuiltinCall()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | subgraph_utils.cpp | 221 const auto subgraphOutputs = insertGraph( in unmergeSubgraph()
|
| /aosp_15_r20/external/pytorch/torch/_C/ |
| H A D | __init__.pyi.in | 885 def insertGraph(self, callee: Graph, inputs: List[Value]) -> List[Value]: ...
|