Searched defs:out_invstd (Results 1 – 2 of 2) sorted by relevance
662 const Tensor& out_mean, const Tensor& out_invstd, const Tensor& input_, double epsilon) { in batch_norm_stats_cuda_template()947 accscalar_t* __restrict__ out_invstd, in batch_norm_collect_statistics_channels_last_kernel()1427 const Tensor& out_mean, const Tensor& out_invstd, const Tensor& input, double epsilon) { in batch_norm_stats_channels_last_cuda_template()
417 void batch_norm_calc_invstd(const Tensor& out_invstd, const Tensor& running_var, double epsilon) { in batch_norm_calc_invstd()