Searched refs:batch_norm_gather_stats_with_counts_cuda (Results 1 – 1 of 1) sorted by relevance
702 …return batch_norm_gather_stats_with_counts_cuda(self, mean, invstd, running_mean, running_var, mom… in batch_norm_gather_stats_cuda()706 std::tuple<Tensor, Tensor> batch_norm_gather_stats_with_counts_cuda( in batch_norm_gather_stats_with_counts_cuda() function