Home
last modified time | relevance | path

Searched refs:manage_output_tensors_enabled_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.cpp912 manage_output_tensors_enabled_( in BlockRunner()
1078 manage_output_tensors_enabled_, in create_memory_planner()
1305 DCHECK(!manage_output_tensors_enabled_ || checkOutputTensorMemoryLeaks()); in run_impl()
1591 if (manage_output_tensors_enabled_) { in benchmark_model()
1602 if (manage_output_tensors_enabled_) { in benchmark_model()
1966 if (!planner_ || !manage_output_tensors_enabled_) { in isManagedOutputTensorValue()
1974 if (!manage_output_tensors_enabled_) { in disableManageOutputTensors()
1977 manage_output_tensors_enabled_ = false; in disableManageOutputTensors()
H A Dimpl.h796 bool manage_output_tensors_enabled_ = false; variable