Searched defs:mean_p (Results 1 – 3 of 3) sorted by relevance
192 const int mean_p = (lower.p + upper.p) >> 1; in read_training_per_lane() local542 const int mean_p = (lower.p + upper.p) >> 1; in write_training_per_group() local
358 const float* mean_p = &inputs(i, 0); in BroadcastAcrossImage() local
6105 const Tensor& mean_p, in _invstd_jvp()6130 const Tensor& mean_p, in _norm_jvp()6212 Tensor mean_p; in batch_norm_jvp() local6268 auto mean_p = saved_mean.view(view_size); in layer_norm_jvp() local6336 const Tensor& mean_p, in group_norm_mean_jvp()6347 const Tensor& mean_p, in group_norm_invstd_jvp()