Searched refs:y_backprop_rest_by_depth (Results 1 – 2 of 2) sorted by relevance
84 auto y_backprop_rest_by_depth = in operator ()() local88 offset_backprop.device(d) = y_backprop_rest_by_depth.sum(reduction_axis); in operator ()()95 (y_backprop_rest_by_depth * in operator ()()101 (y_backprop_rest_by_depth * in operator ()()
464 auto y_backprop_rest_by_depth = in operator ()() local470 scratch_tensor.device(d) = y_backprop_rest_by_depth * x_scaled; in operator ()()486 y_backprop_rest_by_depth - y_backprop_mean_rest_by_depth; in operator ()()491 scratch_tensor.device(d) = y_backprop_rest_by_depth * x_centered; in operator ()()584 auto y_backprop_rest_by_depth = in operator ()() local596 y_backprop_rest_by_depth * in operator ()()602 (y_backprop_rest_by_depth * in operator ()()