Home
last modified time | relevance | path

Searched defs:batch_norm_backward_elemt_cuda_template (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DNormalization.cuh838 Tensor batch_norm_backward_elemt_cuda_template(const Tensor& grad_out_, const Tensor& input_, in batch_norm_backward_elemt_cuda_template() function
889 Tensor batch_norm_backward_elemt_cuda_template(const Tensor& grad_out_, const Tensor& input_, in batch_norm_backward_elemt_cuda_template() function