Searched refs:batch_norm_backward_cuda_template (Results 1 – 2 of 2) sorted by relevance
590 return batch_norm_backward_cuda_template<scalar_t, accscalar_t, int32_t>( in batch_norm_backward_cuda()594 return batch_norm_backward_cuda_template<scalar_t, scalar_t, int32_t>( in batch_norm_backward_cuda()
603 std::tuple<Tensor, Tensor, Tensor> batch_norm_backward_cuda_template(const Tensor& grad_out_, const… in batch_norm_backward_cuda_template() function