Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_utils.cc295 for (bool manage_output_tensors : {true, false}) { in testStaticRuntime()
297 if (!enable_out_variant && manage_output_tensors) { in testStaticRuntime()
307 opts.manage_output_tensors = manage_output_tensors; in testStaticRuntime()
322 VLOG(2) << "manage_output_tensors: " << manage_output_tensors; in testStaticRuntime()
330 if (manage_output_tensors) { in testStaticRuntime()
348 if (manage_output_tensors) { in testStaticRuntime()
369 if (manage_output_tensors) { in testStaticRuntime()
382 if (manage_output_tensors) { in testStaticRuntime()
390 if (manage_output_tensors) { in testStaticRuntime()
H A Dtest_static_module.cc622 for (auto manage_output_tensors : {true, false}) { in TEST()
623 if (manage_output_tensors && !enable_out_variant) { in TEST()
635 << ", manage_output_tensors: " << manage_output_tensors; in TEST()
637 enable_out_variant, optimize_memory, manage_output_tensors}; in TEST()
660 if (manage_output_tensors) { in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dmemory_planner.cpp207 bool manage_output_tensors) { in MemoryPlanner() argument
228 (manage_output_tensors && in MemoryPlanner()
269 if (enable_out_variant && manage_output_tensors) { in MemoryPlanner()
376 bool manage_output_tensors, in StandardMemoryPlanner() argument
382 manage_output_tensors) { in StandardMemoryPlanner()
H A Dimpl.cpp297 << opts.manage_output_tensors << ", use_copy_variants " in PrepareForStaticModule()
558 if (opts.manage_output_tensors) { in StaticModule()
825 if (opts.manage_output_tensors && is_tensor_type && in prepare_for_memory_planner()
913 is_root_block_ && sm.opts().manage_output_tensors), in BlockRunner()
1727 bool manage_output_tensors = static_module_.opts().manage_output_tensors; in benchmark_individual_ops() local
1744 if (manage_output_tensors) { in benchmark_individual_ops()
1755 if (manage_output_tensors) { in benchmark_individual_ops()
1787 if (manage_output_tensors) { in benchmark_individual_ops()
1917 if (!static_module_.opts().manage_output_tensors) { in deallocateOutputTensors()
1930 if (!static_module_.opts().manage_output_tensors || !planner_) { in checkOutputTensorMemoryLeaks()
H A Dmemory_planner.h150 bool manage_output_tensors);
288 bool manage_output_tensors,
H A DREADME.md104 `StaticRuntime` can optionally manage output tensors via the `manage_output_tensors` option in `Sta…
H A Dimpl.h157 bool manage_output_tensors{false};