Searched refs:batch_norm_backward_reduce_cuda (Results 1 – 1 of 1) sorted by relevance
629 batch_norm_backward_reduce_cuda( in batch_norm_backward_cuda()725 std::tuple<Tensor, Tensor, Tensor, Tensor> batch_norm_backward_reduce_cuda(const Tensor& grad_outpu… in batch_norm_backward_reduce_cuda() function