Home
last modified time | relevance | path

Searched refs:StaticModuleOptions (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.h148 struct TORCH_API StaticModuleOptions { struct
186 explicit StaticRuntimeMetadata(const StaticModuleOptions& opts) in StaticRuntimeMetadata()
189 const StaticModuleOptions& get_opts() { in get_opts()
194 StaticModuleOptions opts_;
372 const StaticModuleOptions& opt);
409 const StaticModuleOptions& opts = StaticModuleOptions(),
415 const StaticModuleOptions& opts = StaticModuleOptions(),
422 const StaticModuleOptions& opts);
442 const StaticModuleOptions& opts() const;
538 StaticModuleOptions opts_;
H A Dimpl.cpp158 const StaticModuleOptions& opts, in OptimizeGraph()
270 const StaticModuleOptions& opts, in PrepareGraphForStaticModule()
292 const StaticModuleOptions& opts, in PrepareForStaticModule()
321 const StaticModuleOptions& opts, in PrepareForStaticModule()
530 const StaticModuleOptions& opts, in StaticModule()
539 const StaticModuleOptions& opts, in StaticModule()
548 const StaticModuleOptions& opts) in StaticModule()
804 const StaticModuleOptions& opts) { in prepare_for_memory_planner()
856 const StaticModuleOptions& StaticModule::opts() const { in opts()
H A DREADME.md104 …n optionally manage output tensors via the `manage_output_tensors` option in `StaticModuleOptions`.
193 see `StaticModuleOptions` for more details.
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_cpu_fusion.cc21 StaticModuleOptions opts; // start with the defaults. in TEST()
56 StaticModuleOptions opts; // start with the defaults. in TEST()
95 StaticModuleOptions opts; // start with the defaults. in TEST()
H A Dtest_utils.cc68 const StaticModuleOptions& opt) const = 0;
82 StaticModule makeStaticModule(const StaticModuleOptions& opt) const override { in makeStaticModule()
105 StaticModule makeStaticModule(const StaticModuleOptions& opt) const override { in makeStaticModule()
304 StaticModuleOptions opts; in testStaticRuntime()
H A Dtest_static_module.cc636 torch::jit::StaticModuleOptions opts{ in TEST()
701 torch::jit::StaticModuleOptions opts{ in TEST()
748 torch::jit::StaticModuleOptions opts{ in TEST()
772 torch::jit::StaticModuleOptions opts{ in TEST()
809 torch::jit::StaticModuleOptions opts{ in TEST()
H A Ddeep_wide_pt_bench.cc166 torch::jit::StaticModuleOptions opts; in BM_long_static_memory_optimization()