Searched refs:supportsDepthwiseConvolutionWithCuDNN (Results 1 – 4 of 4) sorted by relevance
40 bool supportsDepthwiseConvolutionWithCuDNN() const override;
278 bool CUDAHooks::supportsDepthwiseConvolutionWithCuDNN() const { in supportsDepthwiseConvolutionWithCuDNN() function in at::cuda::detail::CUDAHooks
144 virtual bool supportsDepthwiseConvolutionWithCuDNN() const { in supportsDepthwiseConvolutionWithCuDNN() function
466 if (detail::getCUDAHooks().supportsDepthwiseConvolutionWithCuDNN()) { in use_cudnn_depthwise()