Searched refs:cudnnv8_enabled_check_debug (Results 1 – 3 of 3) sorted by relevance
1186 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()
80 inline bool cudnnv8_enabled_check_debug() { in cudnnv8_enabled_check_debug() function
429 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()