Searched refs:rest_size_inv (Results 1 – 1 of 1) sorted by relevance
177 U rest_size_inv = static_cast<U>(1.0f / static_cast<U>(rest_size)); in operator ()() local185 batch_mean.device(d) = (x_rest_by_depth.sum(reduce_dims) * rest_size_inv); in operator ()()190 x_centered.square().sum(reduce_dims) * rest_size_inv; in operator ()()421 U rest_size_inv = static_cast<U>(1.0f / static_cast<U>(rest_size)); in operator ()() local482 y_backprop_sum_one_by_depth * rest_size_inv; in operator ()()