Searched defs:invstd_nd (Results 1 – 1 of 1) sorted by relevance
215 auto invstd_nd = as_nd(invstd); in batch_norm_elementwise_backward_train() local262 auto invstd_nd = invstd.as_strided(shape, strides); in batch_norm_elementwise_backward_eval() local