HomeSort by relevance Sort by last modified time
    Searched refs:y_backprop_rest_by_depth (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.h 102 auto y_backprop_rest_by_depth = local
106 offset_backprop.device(d) = y_backprop_rest_by_depth.sum(reduction_axis);
113 (y_backprop_rest_by_depth *
119 (y_backprop_rest_by_depth * ((scratch1 * scale)
fused_batch_norm_op.cc 191 auto y_backprop_rest_by_depth = local
194 (y_backprop_rest_by_depth * x_scaled).sum(reduce_dims);
195 auto y_backprop_sum = y_backprop_rest_by_depth.sum(reduce_dims);
205 y_backprop_rest_by_depth - y_backprop_mean_rest_by_depth;
209 (y_backprop_rest_by_depth * x_centered).mean(reduce_dims))
    [all...]

Completed in 1975 milliseconds