Searched refs:PythonOp (Results 1 – 18 of 18) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_ir.h | 12 struct ConcretePythonOp : public PythonOp {
15 ConcretePythonOp(Graph* graph) : PythonOp(graph, ::c10::prim::PythonOp) {} in ConcretePythonOp()
|
| H A D | python_interpreter.cpp | 79 prim::PythonOp,
|
| H A D | python_ir.cpp | 31 Symbol ConcretePythonOp::Kind = prim::PythonOp;
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | inplace_check.cpp | 7 if (node->kind() == prim::PythonOp && node->hasAttribute(attr::inplace)) { in CheckInplace()
10 std::string("inplace ") + static_cast<PythonOp*>(node)->name() + in CheckInplace()
|
| H A D | lower_graph.cpp | 82 if (e.n->kind() == prim::PythonOp) { in lower_graph()
|
| H A D | lower_tuples.cpp | 29 prim::PythonOp,
|
| H A D | onnx.cpp | 625 } else if (k == prim::PythonOp) { in NodeToONNX()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | operator.cpp | 212 prim::ListUnpack, prim::Print, prim::PythonOp, in printerHasSpecialCaseFor()
308 prim::PythonOp, in aliasAnalysisHasSpecialCaseFor()
|
| H A D | graph_executor.cpp | 896 if (n->kind() == prim::PythonOp) in mayIntroduceGradient()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/ |
| H A D | autograd_function_process.cpp | 14 if (node->kind() == prim::PythonOp) { in convertSubgraphToSubBlock()
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | onnx_torchscript.rst | 438 .. PythonOp Symbolic
446 .. All autograd ``Function``\ s appear in the TorchScript graph as ``prim::PythonOp`` nodes.
489 .. return _unimplemented("prim::PythonOp", "unknown node kind: " + name)
495 .. . register_custom_op_symbolic("prim::PythonOp", symbolic_python_op, 1)
501 where a function to register ``prim::PythonOp`` as custom symbolic functions is not provided,
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | ir.cpp | 313 if (kind() == prim::PythonOp) { in print()
314 auto* pyOp = static_cast<const ::torch::jit::PythonOp*>(this); in print()
490 case prim::PythonOp: { in lint()
492 auto* value = static_cast<const PythonOp*>(this); in lint()
1163 case prim::PythonOp: in hasSideEffects()
|
| H A D | alias_analysis.cpp | 695 case prim::PythonOp: in analyzeImpl()
|
| H A D | ir.h | 1582 struct TORCH_API PythonOp : public Node { struct
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | list_model_parameters.cpp | 87 if (use.user->kind() == prim::PythonOp) in getParamAttributes()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | interned_strings.h | 74 _(prim, PythonOp) \
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | export.cpp | 139 if (node->kind() == prim::PythonOp) { in validateBlock()
142 auto py_node = static_cast<PythonOp*>(node); in validateBlock()
|
| H A D | python_print.cpp | 999 case prim::PythonOp: { in printRHS()
1000 auto value = static_cast<const PythonOp*>(node); in printRHS()
|