| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_static_runtime.py | 10 from torch.testing._internal.static_module import StaticModule
203 static_runtime_module = StaticModule(torch_graph)
216 static_runtime_module = StaticModule(torch_graph)
230 static_runtime_module = StaticModule(torch_graph)
243 static_runtime_module = StaticModule(torch_graph)
257 static_runtime_module = StaticModule(torch_graph)
270 static_runtime_module = StaticModule(torch_graph)
285 static_runtime_module = StaticModule(torch_graph)
300 static_runtime_module = StaticModule(torch_graph)
321 static_runtime_module = StaticModule(torch_graph)
[all …]
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_module.cc | 23 StaticModule makeStaticModuleFromScript(const std::string& script) { in makeStaticModuleFromScript()
26 return StaticModule(m); in makeStaticModuleFromScript()
90 TEST(StaticModule, ValueGroup) { in TEST() argument
105 torch::jit::StaticModule sm(input_graph); in TEST()
128 TEST(StaticModule, IsOptimizableContainerType_NonOptimizableInputs) { in TEST() argument
149 TEST(StaticModule, IsOptimizableContainerType_WrongType) { in TEST() argument
171 TEST(StaticModule, IsOptimizableContainerType_CanUseOutVariant) { in TEST() argument
195 TEST(StaticModule, RValueInputs) { in TEST() argument
459 torch::jit::StaticModule smod(mod); in TEST()
478 torch::jit::StaticModule smod(mod); in TEST()
[all …]
|
| H A D | deep_wide_pt_bench.cc | 78 torch::jit::StaticModule smod(mod); in BM_deep_wide_static()
93 std::shared_ptr<torch::jit::StaticModule> getStaticModule() { in getStaticModule()
95 std::make_shared<torch::jit::StaticModule>(getDeepAndWideSciptModel()); in getStaticModule()
118 torch::jit::StaticModule smod(mod); in BM_leaky_relu_const()
132 torch::jit::StaticModule smod(mod); in BM_leaky_relu()
150 torch::jit::StaticModule smod(mod); in BM_signed_log1p()
168 torch::jit::StaticModule smod(mod, false, opts); in BM_long_static_memory_optimization()
|
| H A D | test_utils.cc | 67 virtual StaticModule makeStaticModule(
82 StaticModule makeStaticModule(const StaticModuleOptions& opt) const override { in makeStaticModule()
83 return torch::jit::StaticModule( in makeStaticModule()
105 StaticModule makeStaticModule(const StaticModuleOptions& opt) const override { in makeStaticModule()
106 return StaticModule(graph_, opt, /* sample_inputs */ {}); in makeStaticModule()
223 Node* getNodeWithKind(const StaticModule& smodule, const std::string& kind) { in getNodeWithKind()
238 bool hasNodeWithKind(const StaticModule& smodule, const std::string& kind) { in hasNodeWithKind()
406 torch::jit::StaticModule& smodule, in hasProcessedNodeWithName()
|
| H A D | test_utils.h | 22 class StaticModule; variable
41 torch::jit::StaticModule& smodule,
46 Node* getNodeWithKind(const StaticModule& smodule, const std::string& kind);
49 bool hasNodeWithKind(const StaticModule& smodule, const std::string& kind);
|
| H A D | test_static_runtime.cc | 1812 torch::jit::StaticModule smod(mod); in TEST()
2601 torch::jit::StaticModule smodule(graph); in TEST()
2622 torch::jit::StaticModule smodule(graph); in TEST()
2667 torch::jit::StaticModule smodule(graph); in TEST()
2689 torch::jit::StaticModule smodule(graph); in TEST()
2782 torch::jit::StaticModule smodule(module); in TEST()
2815 torch::jit::StaticModule smodule(module); in TEST()
2831 torch::jit::StaticModule smodule(module); in TEST()
2860 torch::jit::StaticModule smodule(module); in TEST()
3105 auto static_module = StaticModule(graph); in TEST()
[all …]
|
| H A D | test_cpu_fusion.cc | 27 auto smodule = StaticModule(m, /* is_frozen */ false, opts, {input1, input2}); in TEST()
61 auto smodule = StaticModule( in TEST()
100 auto smodule = StaticModule( in TEST()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | init.cpp | 16 py::class_<StaticModule> static_module(m, "StaticModule"); in initStaticModuleBindings()
52 [](StaticModule& self, in initStaticModuleBindings()
72 [](StaticModule& self, in initStaticModuleBindings()
85 [](StaticModule& self, in initStaticModuleBindings()
98 [](StaticModule& self, in initStaticModuleBindings()
123 return StaticModule(g); in initStaticModuleBindings()
127 [](const torch::jit::Module& module) { return StaticModule(module); }) in initStaticModuleBindings()
|
| H A D | impl.cpp | 528 StaticModule::StaticModule( in StaticModule() function in torch::jit::StaticModule
532 : StaticModule( in StaticModule()
536 StaticModule::StaticModule( in StaticModule() function in torch::jit::StaticModule
541 : StaticModule( in StaticModule()
545 StaticModule::StaticModule( in StaticModule() function in torch::jit::StaticModule
616 size_t StaticModule::prepareBlockInfo( in prepareBlockInfo()
666 void StaticModule::attachNodeMetadata(Block* block) { in attachNodeMetadata()
677 void StaticModule::prepareFunctionsAndConstants( in prepareFunctionsAndConstants()
709 size_t StaticModule::prepareStaticNodeInfos( in prepareStaticNodeInfos()
856 const StaticModuleOptions& StaticModule::opts() const { in opts()
[all …]
|
| H A D | README.md | 68 **In both modes, `StaticRuntime` may not be used after its associated `StaticModule` is destructed!…
165 StaticModule◄───────────StaticRuntime───────────►BlockRunner────────►MemoryPlanner │ …
178 ### `StaticModule` and `StaticRuntime`
180 `StaticModule`s are constructed from `torch::jit::Module`s and can be used to construct `StaticRunt…
181 instances. Each `StaticModule` caches exactly one `StaticRuntime` instance - it is lazily initializ…
184 `StaticModule::operator()` can be used directly to make predictions. Under the hood, this method ju…
186 `StaticModule::operator()` is not thread-safe.
192 `StaticModule` takes a set of options that control the behavior of the runtime instances that it sp…
224 upon `StaticModule` construction according to the out variant/native/JIT fallback lookup rules desc…
|
| H A D | impl.h | 405 class TORCH_API StaticModule {
407 explicit StaticModule(
412 explicit StaticModule(
419 explicit StaticModule(
582 const StaticModule& sm,
787 const StaticModule& static_module_;
1047 explicit StaticRuntime(const StaticModule& sm);
|
| H A D | static_method.h | 11 std::shared_ptr<StaticModule> static_module, in StaticMethod()
44 std::shared_ptr<StaticModule> static_module_;
|
| H A D | native_ops.cpp | 1060 std::shared_ptr<StaticModule> smodule, in ForkedSubgraphSRLauncher()
1082 std::shared_ptr<StaticModule> smodule_;
1128 std::make_shared<StaticModule>(forkedGraph, sr_metadata->get_opts()); in __anon75e5f0516602()
|
| H A D | fusion.cpp | 43 auto module = std::make_shared<torch::jit::StaticModule>(g); in createStaticSubgraphRuntime()
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/python3/lib/python3.11/site-packages/setuptools/config/ |
| D | expand.py | 63 class StaticModule: class
191 return getattr(StaticModule(module_name, spec), attr_name)
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/python3/lib/python3.11/site-packages/setuptools/config/ |
| D | expand.py | 63 class StaticModule: class
191 return getattr(StaticModule(module_name, spec), attr_name)
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/python3/lib/python3.11/site-packages/setuptools/config/ |
| D | expand.py | 63 class StaticModule: class
191 return getattr(StaticModule(module_name, spec), attr_name)
|
| /aosp_15_r20/external/python/setuptools/setuptools/config/ |
| D | expand.py | 57 class StaticModule: class
191 return getattr(StaticModule(module_name, spec), attr_name)
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/python3/lib/python3.11/site-packages/setuptools/config/ |
| D | expand.py | 63 class StaticModule: class
191 return getattr(StaticModule(module_name, spec), attr_name)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | static_module.py | 7 class StaticModule: class
|