Searched refs:batch_norm_calc_invstd (Results 1 – 1 of 1) sorted by relevance
417 void batch_norm_calc_invstd(const Tensor& out_invstd, const Tensor& running_var, double epsilon) { in batch_norm_calc_invstd() function448 batch_norm_calc_invstd(save_invstd, save_invstd, epsilon); in batch_norm_cuda_out()454 batch_norm_calc_invstd(save_invstd, running_var_opt.value(), epsilon); in batch_norm_cuda_out()623 batch_norm_calc_invstd(invstd, *running_var, epsilon); in batch_norm_backward_cuda()