Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/cuda/detail/
DCUDAHooks.h40 bool supportsDepthwiseConvolutionWithCuDNN() const override;
DCUDAHooks.cpp278 bool CUDAHooks::supportsDepthwiseConvolutionWithCuDNN() const { in supportsDepthwiseConvolutionWithCuDNN() function in at::cuda::detail::CUDAHooks
/external/pytorch/aten/src/ATen/detail/
DCUDAHooksInterface.h144 virtual bool supportsDepthwiseConvolutionWithCuDNN() const { in supportsDepthwiseConvolutionWithCuDNN() function
/external/pytorch/aten/src/ATen/native/
DConvolution.cpp466 if (detail::getCUDAHooks().supportsDepthwiseConvolutionWithCuDNN()) { in use_cudnn_depthwise()