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