Home
last modified time | relevance | path

Searched refs:checkOutputTensorMemoryLeaks (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_utils.cc333 runtime.checkOutputTensorMemoryLeaks(); in testStaticRuntime()
351 runtime.checkOutputTensorMemoryLeaks(); in testStaticRuntime()
372 runtime.checkOutputTensorMemoryLeaks(); in testStaticRuntime()
385 runtime.checkOutputTensorMemoryLeaks(); in testStaticRuntime()
393 runtime.checkOutputTensorMemoryLeaks(); in testStaticRuntime()
H A Dtest_static_module.cc662 runtime.checkOutputTensorMemoryLeaks(); in TEST()
723 runtime.checkOutputTensorMemoryLeaks(); in TEST()
739 runtime.checkOutputTensorMemoryLeaks(); in TEST()
763 runtime.checkOutputTensorMemoryLeaks(); in TEST()
789 EXPECT_THROW(runtime.checkOutputTensorMemoryLeaks(), std::exception); in TEST()
794 runtime.checkOutputTensorMemoryLeaks(); in TEST()
824 runtime.checkOutputTensorMemoryLeaks(); in TEST()
833 runtime.checkOutputTensorMemoryLeaks(); in TEST()
851 runtime.checkOutputTensorMemoryLeaks(); in TEST()
870 runtime.checkOutputTensorMemoryLeaks(); in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.cpp1305 DCHECK(!manage_output_tensors_enabled_ || checkOutputTensorMemoryLeaks()); in run_impl()
1925 DCHECK(checkOutputTensorMemoryLeaks()); in deallocateOutputTensors()
1929 bool BlockRunner::checkOutputTensorMemoryLeaks() { in checkOutputTensorMemoryLeaks() function in torch::jit::BlockRunner
2302 bool StaticRuntime::checkOutputTensorMemoryLeaks() { in checkOutputTensorMemoryLeaks() function in torch::jit::StaticRuntime
2303 return block_->checkOutputTensorMemoryLeaks(); in checkOutputTensorMemoryLeaks()
H A Dimpl.h686 bool checkOutputTensorMemoryLeaks();
1072 bool checkOutputTensorMemoryLeaks();