Searched defs:rest_by_depth (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fused_batch_norm_op.cc | 170 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()() local
313 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()() local
429 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()() local
576 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()() local
|
| H A D | fused_batch_norm_op.cu.cc | 68 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()() local
|