Searched refs:rest_by_one (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_norm_op.h | 40 Eigen::DSizes<int, 2> rest_by_one(rest_size, 1); in operator()
44 Eigen::IndexList<int, Eigen::type2index<1> > rest_by_one; in operator() local
45 rest_by_one.set(0, rest_size); in operator()
54 mean.reshape(one_by_depth).broadcast(rest_by_one)) * in operator()
58 .broadcast(rest_by_one) + in operator()
59 beta.reshape(one_by_depth).broadcast(rest_by_one); in operator()
63 mean.reshape(one_by_depth).broadcast(rest_by_one)) * in operator()
67 .broadcast(rest_by_one) + in operator()
68 beta.reshape(one_by_depth).broadcast(rest_by_one); in operator()
92 Eigen::DSizes<Index, 2> rest_by_one(rest_size, 1); in operator()
[all …]
|
| D | fused_batch_norm_op.cu.cc | 71 Eigen::array<int, 2> rest_by_one({rest_size, 1}); in operator ()() local
76 Eigen::IndexList<Eigen::Index, Eigen::type2index<1> > rest_by_one; in operator ()() local
77 rest_by_one.set(0, rest_size); in operator ()()
97 pop_mean.reshape(one_by_depth).broadcast(rest_by_one))) in operator ()()
102 ((scratch1 * scale).reshape(one_by_depth).broadcast(rest_by_one))) in operator ()()
|
| D | fused_batch_norm_op.cc | 564 Eigen::array<int, 2> rest_by_one({rest_size, 1}); in operator ()() local
568 Eigen::IndexList<Eigen::Index, Eigen::type2index<1>> rest_by_one; in operator ()() local
569 rest_by_one.set(0, rest_size); in operator ()()
598 pop_mean.reshape(one_by_depth).broadcast(rest_by_one)); in operator ()()
604 .broadcast(rest_by_one))) in operator ()()
|