Home
last modified time | relevance | path

Searched refs:_cudnn_set_conv_benchmark_empty_cache (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DConvUtils.h352 TORCH_API void _cudnn_set_conv_benchmark_empty_cache(bool enable);
DConvolution.cpp2281 void _cudnn_set_conv_benchmark_empty_cache(bool enable) { in _cudnn_set_conv_benchmark_empty_cache() function
/external/pytorch/torch/_inductor/
Dcudagraph_trees.py167 torch._C._cudnn_set_conv_benchmark_empty_cache(False)
171 torch._C._cudnn_set_conv_benchmark_empty_cache(prev)
/external/pytorch/torch/csrc/cuda/
DModule.cpp1049 return at::native::_cudnn_set_conv_benchmark_empty_cache(enable); in registerCudaDeviceProperties()
/external/pytorch/torch/_C/
D__init__.pyi.in1848 def _cudnn_set_conv_benchmark_empty_cache(enable: _bool) -> None: ...