| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_sugared_value.h | 26 GraphFunction& m,
49 GraphFunction& m,
58 GraphFunction& m,
63 GraphFunction& m,
66 Value* asValue(const SourceRange& loc, GraphFunction& m) override { in asValue()
93 GraphFunction& m,
105 GraphFunction& m,
117 GraphFunction& caller, in call()
138 GraphFunction& f, in call()
171 Value* asValue(const SourceRange& loc, GraphFunction& m) override;
[all …]
|
| H A D | python_sugared_value.cpp | 118 GraphFunction& m, in call()
172 GraphFunction& m, in asTuple()
182 GraphFunction& m, in attr()
210 GraphFunction& m, in attr()
221 GraphFunction& m, in attr()
261 Value* ModuleValue::asValue(const SourceRange& loc, GraphFunction& m) { in asValue()
267 GraphFunction& m) { in asTupleValue()
303 GraphFunction& m, in getitem()
386 GraphFunction& m, in checkInterface()
399 GraphFunction& m, in recurseThroughNestedModules()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | sugared_value.h | 34 virtual Value* asValue(const SourceRange& loc, GraphFunction& m) { in asValue()
41 GraphFunction& m, in attr()
48 GraphFunction& m, in hasAttr()
56 GraphFunction& m, in setAttr()
68 GraphFunction& m,
76 GraphFunction& m) { in asTupleValue()
89 GraphFunction& m, in call()
115 GraphFunction& m) { in iter()
137 virtual Value* len(const SourceRange& loc, GraphFunction& m) { in len()
146 GraphFunction& m,
[all …]
|
| H A D | sugared_value.cpp | 20 GraphFunction& m, in call()
51 GraphFunction& m, in call()
71 GraphFunction& m, in hasAttr()
101 GraphFunction& m, in attr()
287 GraphFunction& m, in asTuple()
333 GraphFunction& m, in setAttr()
417 GraphFunction& m, in call()
455 Value* SimpleValue::len(const SourceRange& loc, GraphFunction& m) { in len()
472 GraphFunction& m, in getitem()
510 SugaredValuePtr SimpleValue::iter(const SourceRange& loc, GraphFunction& m) { in iter()
[all …]
|
| H A D | resolver.h | 34 GraphFunction& m, in resolveValue()
49 GraphFunction& m, in resolveValue()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/api/ |
| H A D | function_impl.cpp | 23 c10::FunctionSchema defaultSchemaFor(const GraphFunction& function) { in defaultSchemaFor()
64 static void placeholderCreator(GraphFunction&) { in placeholderCreator() argument
68 void GraphFunction::run(Stack& stack) { in run()
73 c10::intrusive_ptr<c10::ivalue::Future> GraphFunction::runAsync( in runAsync()
79 void GraphFunction::ensure_defined() { in ensure_defined()
89 const c10::FunctionSchema& GraphFunction::getSchema() const { in getSchema()
96 std::shared_ptr<Graph> GraphFunction::optimized_graph() const { in optimized_graph()
112 GraphFunction::SpecializationKey GraphFunction::currentSpecialization() const { in currentSpecialization()
161 GraphFunction* tryToGraphFunction(Function& function) noexcept { in tryToGraphFunction()
162 return tryToGraphFunctionImpl<GraphFunction>(function); in tryToGraphFunction()
[all …]
|
| H A D | function_impl.h | 9 struct TORCH_API GraphFunction : public Function { struct
10 GraphFunction(
13 std::function<void(GraphFunction&)> function_creator,
27 std::function<void(GraphFunction&)> function_creator() const { in function_creator() argument
167 std::function<void(GraphFunction&)> function_creator_; argument
176 TORCH_API GraphFunction* tryToGraphFunction(Function&) noexcept;
177 TORCH_API GraphFunction& toGraphFunction(Function&);
178 TORCH_API const GraphFunction& toGraphFunction(const Function&);
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_node_lowering.cpp | 21 std::shared_ptr<torch::jit::GraphFunction> function, in LowerBuiltin()
28 std::shared_ptr<torch::jit::GraphFunction> function, in LowerBuiltin()
35 std::shared_ptr<torch::jit::GraphFunction> function, in LowerTSBuiltin()
59 std::shared_ptr<torch::jit::GraphFunction> function) { in GenerateClone()
71 std::shared_ptr<torch::jit::GraphFunction> function, in Lower()
82 std::shared_ptr<torch::jit::GraphFunction> function, in Lower()
91 std::shared_ptr<torch::jit::GraphFunction> function, in Lower()
104 std::shared_ptr<torch::jit::GraphFunction> function, in Lower()
122 std::shared_ptr<torch::jit::GraphFunction> function, in Lower()
|
| H A D | ts_node.h | 56 std::shared_ptr<torch::jit::GraphFunction> function,
101 std::shared_ptr<torch::jit::GraphFunction> function,
|
| H A D | ts_lowering_context.cpp | 15 std::make_shared<torch::jit::GraphFunction>(name, graph_, nullptr)) {} in TSLoweringContext()
25 std::make_shared<torch::jit::GraphFunction>(name, graph_, nullptr)) { in TSLoweringContext()
|
| H A D | ts_node_lowering.h | 11 std::shared_ptr<torch::jit::GraphFunction> function,
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | graph_function.cc | 20 GraphFunction::GraphFunction(FunctionDef fdef) in GraphFunction() function in tensorflow::tracing::graph::GraphFunction
22 GraphFunction::~GraphFunction() {} in ~GraphFunction()
23 Status GraphFunction::GetFunctionDef(FunctionDef** fdef) { in GetFunctionDef()
|
| H A D | graph_function.h | 24 class GraphFunction : public AbstractFunction {
26 explicit GraphFunction(FunctionDef fdef);
27 ~GraphFunction() override;
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_exception.cpp | 28 torch::jit::GraphFunction* gf = in TEST()
29 (torch::jit::GraphFunction*)&cu_ptr->get_function("foo"); in TEST()
65 torch::jit::GraphFunction& caller, in call()
87 torch::jit::GraphFunction& m, in resolveValue()
139 torch::jit::GraphFunction* gf = in TEST()
140 (torch::jit::GraphFunction*)&cu->get_function("foo"); in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libtf/tests/ |
| H A D | function_test.cc | 44 using tensorflow::tracing::graph::GraphFunction;
154 AbstractFunctionPtr trace(new GraphFunction(fdef), /*add_ref=*/false); in TEST_P()
172 AbstractFunctionPtr trace(new GraphFunction(fdef), /*add_ref=*/false); in TEST_P()
195 AbstractFunctionPtr trace(new GraphFunction(fdef), /*add_ref=*/false); in TEST_P()
218 AbstractFunctionPtr trace(new GraphFunction(fdef), /*add_ref=*/false); in TEST_P()
240 AbstractFunctionPtr trace(new GraphFunction(fdef), /*add_ref=*/false); in TEST_P()
265 AbstractFunctionPtr trace(new GraphFunction(fdef), /*add_ref=*/false); in TEST_P()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | export_bytecode.cpp | 95 static std::vector<std::unique_ptr<GraphFunction>> inlineFunctions( in inlineFunctions()
104 std::vector<std::unique_ptr<GraphFunction>> inlined_functions; in inlineFunctions()
128 auto inlined_func = std::make_unique<GraphFunction>( in inlineFunctions()
250 const GraphFunction& function, in convertJitFunctionToMobileFunction()
|
| H A D | import_source.cpp | 21 GraphFunction& m, in attr()
44 GraphFunction& m,
67 GraphFunction& m, in attr()
227 GraphFunction& m, in resolveValue()
771 GraphFunction& m, in attr()
|
| H A D | export_bytecode.h | 34 const GraphFunction& function,
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/backends/ |
| H A D | backend_resolver.cpp | 18 GraphFunction& m, in attr()
44 GraphFunction& m, in resolveValue()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/interpreter/ |
| H A D | code_impl.h | 96 std::vector<std::unique_ptr<GraphFunction>> forked_functions_;
97 std::vector<std::unique_ptr<GraphFunction>> awaited_functions_;
560 unoptimized_graph](GraphFunction& func) { in emitBailOut()
565 auto func = std::make_unique<GraphFunction>( in emitBailOut()
668 auto forked_fn = std::make_unique<GraphFunction>( in emitFork()
677 auto await_fn = std::make_unique<GraphFunction>( in emitAwaitable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libtf/runtime/ |
| H A D | runtime.cc | 55 using tensorflow::tracing::graph::GraphFunction;
90 AbstractFunctionPtr trace(new GraphFunction(fdef), /*add_ref=*/false); in ImportModule()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | decomposition_registry.cpp | 123 std::optional<GraphFunction*> GetDecompositionFunction( in GetDecompositionFunction()
156 auto new_func = std::make_unique<GraphFunction>( in RegisterDecomposition()
|
| H A D | profiling_graph_executor_impl.cpp | 733 GraphFunction* func, in insertFallbackFunctionCall()
752 static GraphFunction* createFallbackPathFunction( in createFallbackPathFunction()
770 return new GraphFunction(function_name, graph, nullptr); in createFallbackPathFunction()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | inliner.h | 10 TORCH_API GraphFunction* tryToGraphFunction(Node* n);
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | insert_quant_dequant.cpp | 675 std::unique_ptr<GraphFunction> buildGraphFromNodes(
821 std::unordered_map<Value*, std::unique_ptr<GraphFunction>>
961 std::unique_ptr<GraphFunction> SubGraphCloneHelper::buildGraphFromNodes( in buildGraphFromNodes()
965 auto build_observer_graph = [&](GraphFunction& func) { in buildGraphFromNodes()
968 return std::make_unique<GraphFunction>( in buildGraphFromNodes()
1011 std::unique_ptr<GraphFunction> func = in extractAndRunWeightObserver()
|