Home
last modified time | relevance | path

Searched refs:cudnnv8_enabled_check_debug (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConv_v8.cpp1186 if (at::native::cudnnv8_enabled_check_debug()) { in raw_cudnn_convolution_forward_out()
1228 if (at::native::cudnnv8_enabled_check_debug()) { in raw_cudnn_convolution_backward_input_out()
1270 if (at::native::cudnnv8_enabled_check_debug()) { in raw_cudnn_convolution_backward_weight_out()
1315 if (at::native::cudnnv8_enabled_check_debug()) { in raw_cudnn_convolution_add_relu_out()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvUtils.h80 inline bool cudnnv8_enabled_check_debug() { in cudnnv8_enabled_check_debug() function
H A DConvolution.cpp429 if (!(cudnn_version >= 90300 && at::native::cudnnv8_enabled_check_debug())) { in use_cudnn()
440 …DAHooks().supportsBFloat16ConvolutionWithCuDNNv8() && at::native::cudnnv8_enabled_check_debug())) { in use_cudnn()