Searched defs:invstd_p (Results 1 – 1 of 1) sorted by relevance
6106 const Tensor& invstd_p, in _invstd_jvp()6131 const Tensor& invstd_p, in _norm_jvp()6213 Tensor invstd_p; in batch_norm_jvp() local6269 auto invstd_p = saved_invstd.view(view_size); in layer_norm_jvp() local6348 const Tensor& invstd_p, in group_norm_invstd_jvp()