Home
last modified time | relevance | path

Searched refs:one_by_depth (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op.h41 Eigen::DSizes<int, 2> one_by_depth(1, depth); in operator()
46 Eigen::IndexList<Eigen::type2index<1>, int> one_by_depth; in operator() local
47 one_by_depth.set(1, depth); in operator()
54 mean.reshape(one_by_depth).broadcast(rest_by_one)) * in operator()
57 .reshape(one_by_depth) 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()
66 .reshape(one_by_depth) in operator()
68 beta.reshape(one_by_depth).broadcast(rest_by_one); in operator()
93 Eigen::DSizes<Index, 2> one_by_depth(1, depth); in operator()
[all …]
Dfused_batch_norm_op.cc164 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
168 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
169 one_by_depth.set(1, depth); in operator ()()
187 batch_mean.reshape(one_by_depth).broadcast(bcast_spec); in operator ()()
193 .reshape(one_by_depth) in operator ()()
197 (x_scaled + offset.reshape(one_by_depth).broadcast(bcast_spec)) in operator ()()
301 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
305 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
306 one_by_depth.set(1, depth); in operator ()()
314 estimated_mean.reshape(one_by_depth).broadcast(bcast_spec); in operator ()()
[all …]
Dfused_batch_norm_op.cu.cc69 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
73 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
74 one_by_depth.set(1, depth); 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 ()()