Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dmemory_planner.cpp377 bool optimize_memory) in StandardMemoryPlanner() argument
387 if (optimize_memory) { in StandardMemoryPlanner()
H A Dmemory_planner.h289 bool optimize_memory);
H A Dimpl.cpp296 << opts.optimize_memory << ", manage_output_tensors " in PrepareForStaticModule()
563 if (opts_.optimize_memory) { in StaticModule()
1079 static_module_.opts().optimize_memory); in create_memory_planner()
1546 if (static_module_.opts().optimize_memory) { in benchmark()
H A Dimpl.h153 bool optimize_memory{true};
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_module.cc621 for (auto optimize_memory : {true, false}) { in TEST()
628 if (optimize_memory && !enable_out_variant) { in TEST()
634 << ", optimize_memory: " << optimize_memory in TEST()
637 enable_out_variant, optimize_memory, manage_output_tensors}; in TEST()
H A Ddeep_wide_pt_bench.cc167 opts.optimize_memory = state.range(1); in BM_long_static_memory_optimization()
H A Dtest_utils.cc306 opts.optimize_memory = enable_out_variant; in testStaticRuntime()