Searched defs:sum_dy_ (Results 1 – 1 of 1) sorted by relevance
784 Tensor sum_dy_; in batch_norm_backward_reduce_cuda_template() local840 … const Tensor& weight_, const Tensor& sum_dy_, const Tensor& sum_dy_xmu_) { in batch_norm_backward_elemt_cuda_template()891 … const Tensor& weight_, const Tensor& sum_dy_, const Tensor& sum_dy_xmu_, const Tensor& count) { in batch_norm_backward_elemt_cuda_template()