Searched refs:rest_by_depth (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_norm_op.h | 38 Eigen::DSizes<int, 2> rest_by_depth(rest_size, depth); in operator()
52 output.reshape(rest_by_depth).device(d) = in operator()
53 (input.reshape(rest_by_depth) - in operator()
61 output.reshape(rest_by_depth).device(d) = in operator()
62 (input.reshape(rest_by_depth) - in operator()
90 Eigen::DSizes<Index, 2> rest_by_depth(rest_size, depth); in operator()
114 db.device(d) = out_backprop.reshape(rest_by_depth).sum(reduction_axis); in operator()
120 scratch2.device(d) = (out_backprop.reshape(rest_by_depth) * in operator()
121 (input.reshape(rest_by_depth) - in operator()
126 dx.reshape(rest_by_depth).device(d) = in operator()
[all …]
|
| D | fused_batch_norm_op.cc | 161 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()() local
175 auto x_rest_by_depth = x.reshape(rest_by_depth).template cast<U>(); in operator ()()
200 y.reshape(rest_by_depth).device(d) = x_shifted; in operator ()()
298 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()() local
311 auto x_rest_by_depth = x.reshape(rest_by_depth).template cast<U>(); in operator ()()
324 y.reshape(rest_by_depth).device(d) = x_shifted; in operator ()()
408 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()() local
420 auto x_rest_by_depth = x.reshape(rest_by_depth).template cast<U>(); in operator ()()
465 y_backprop.reshape(rest_by_depth).template cast<U>(); in operator ()()
471 redux_sum_u(d, rest_by_depth, scratch_rest_by_depth, scale_backprop_output); in operator ()()
[all …]
|
| D | fused_batch_norm_op.cu.cc | 67 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()() local
85 y_backprop.reshape(rest_by_depth).template cast<U>(); in operator ()()
86 auto input_rest_by_depth = input.reshape(rest_by_depth).template cast<U>(); in operator ()()
100 x_backprop.reshape(rest_by_depth).device(d) = in operator ()()
|