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

  /external/tensorflow/tensorflow/core/kernels/
batch_norm_op.h 40 Eigen::DSizes<int, 2> rest_by_one(rest_size, 1);
44 Eigen::IndexList<int, Eigen::type2index<1> > rest_by_one; local
45 rest_by_one.set(0, rest_size);
54 mean.reshape(one_by_depth).broadcast(rest_by_one)) *
58 .broadcast(rest_by_one) +
59 beta.reshape(one_by_depth).broadcast(rest_by_one);
63 mean.reshape(one_by_depth).broadcast(rest_by_one)) *
67 .broadcast(rest_by_one) +
68 beta.reshape(one_by_depth).broadcast(rest_by_one);
92 Eigen::DSizes<Index, 2> rest_by_one(rest_size, 1)
97 Eigen::IndexList<Index, Eigen::type2index<1> > rest_by_one; local
    [all...]
fused_batch_norm_op.h 89 Eigen::array<int, 2> rest_by_one({rest_size, 1});
94 Eigen::IndexList<Eigen::Index, Eigen::type2index<1> > rest_by_one; local
95 rest_by_one.set(0, rest_size);
115 pop_mean.reshape(one_by_depth).broadcast(rest_by_one)))
122 .broadcast(rest_by_one)))

Completed in 312 milliseconds