Searched refs:batch_norm_use_channels_last_kernels (Results 1 – 1 of 1) sorted by relevance
58 inline bool batch_norm_use_channels_last_kernels(const at::Tensor& self) { in batch_norm_use_channels_last_kernels() function582 !batch_norm_use_channels_last_kernels(input) && in batch_norm_backward_cuda()656 bool use_channels_last_kernel = batch_norm_use_channels_last_kernels(self); in batch_norm_stats_cuda()731 batch_norm_use_channels_last_kernels(grad_output) && in batch_norm_backward_reduce_cuda()732 batch_norm_use_channels_last_kernels(input) && in batch_norm_backward_reduce_cuda()768 batch_norm_use_channels_last_kernels(self) && in batch_norm_backward_elemt_cuda()769 batch_norm_use_channels_last_kernels(input)) { in batch_norm_backward_elemt_cuda()