Searched defs:save_var_transform (Results 1 – 1 of 1) sorted by relevance
202 double momentum, double eps, Tensor& save_mean, Tensor& save_var_transform) { in batch_norm_cpu_update_stats_template()298 Tensor save_var_transform = at::empty({n_input}, input.options().dtype(dtype)); in batch_norm_cpu_update_stats_template() local631 …const Tensor& save_var_transform = c10::value_or_else(save_var_transform_opt, [] {return Tensor();… in _batch_norm_impl_index_backward() local