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 | 173 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
177 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
322 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
326 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
432 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
435 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
584 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
587 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
|
| D | fused_batch_norm_op.cu.cc | 70 Eigen::DSizes<Eigen::Index, 2> one_by_depth(1, depth); in operator ()() local
74 Eigen::IndexList<Eigen::type2index<1>, Eigen::Index> one_by_depth; in operator ()() local
|