Searched defs:one_by_depth (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_norm_op.h | 46 Eigen::IndexList<Eigen::type2index<1>, int> one_by_depth; in operator() local
99 Eigen::IndexList<Eigen::type2index<1>, Index> one_by_depth; in operator() local
|
| D | fused_batch_norm_op.cc | 164 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
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
411 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
414 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
563 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
566 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
|
| D | fused_batch_norm_op.cu.cc | 69 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
|