Searched defs:mean_ptr (Results 1 – 6 of 6) sorted by relevance
139 const float* mean_ptr, const float* variance_ptr, in MinAndMax()195 const uint32_t cols, const float* mean_ptr, in InstanceNorm()
1116 auto mean_ptr = StreamExecutorUtil::AsDeviceMemory<U>(mean); in operator ()() local
166 opmath_t* mean_ptr, in CalcMeanVar()189 opmath_t* mean_ptr, in CalcMeanVar()410 opmath_t* mean_ptr = buffer_ptr + n * 2 * C; in GroupNormKernelImplChannelsLastInternal() local1501 const PT* mean_ptr = mean_data + n * G + g; in GroupNormBackwardKernelImplChannelsLastInternal() local
336 scalar_t* mean_ptr = mean_data + jj_begin; in batch_norm_cpu_collect_stats_channels_last_impl() local361 scalar_t* mean_ptr = mean_data + jj_begin; in batch_norm_cpu_collect_stats_channels_last_impl() local555 const scalar_t* mean_ptr = nullptr; in batch_norm_cpu_backward_channels_last_impl() local
64 float *mean_ptr, float *std_ptr) { in compare_mf()
3207 float* mean_ptr = buffer_ptr + n * 2 * C; local