Searched defs:save_mean_ (Results 1 – 1 of 1) sorted by relevance
604 …const Tensor& running_mean_, const Tensor& running_var_, const Tensor& save_mean_, const Tensor& s… in batch_norm_backward_cuda_template()740 Tensor save_mean_; in batch_norm_gather_stats_cuda_template() local