Searched defs:saved_invstd (Results 1 – 1 of 1) sorted by relevance
6199 const Tensor& saved_invstd, in batch_norm_jvp()6252 const Tensor& saved_invstd, in layer_norm_jvp()6291 const Tensor& saved_invstd, in group_norm_jvp()