Searched defs:out_mean (Results 1 – 1 of 1) sorted by relevance
662 const Tensor& out_mean, const Tensor& out_invstd, const Tensor& input_, double epsilon) { in batch_norm_stats_cuda_template()946 accscalar_t* __restrict__ out_mean, 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()