Searched refs:mean_axes (Results 1 – 1 of 1) sorted by relevance
6986 def _batch_norm_stats(data, memory_format, mean_axes): argument6989 mean_ref = torch.mean(data, mean_axes, keepdim=False)