Searched defs:weight_nd (Results 1 – 1 of 1) sorted by relevance
219 auto weight_nd = weight.defined() ? as_nd(weight) : in batch_norm_elementwise_backward_train() local267 auto weight_nd = weight.as_strided(shape, strides); in batch_norm_elementwise_backward_eval() local