Home
last modified time | relevance | path

Searched refs:rest_size (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op.h36 const int rest_size = input.size() / depth; in operator() local
38 Eigen::DSizes<int, 2> rest_by_depth(rest_size, depth); in operator()
40 Eigen::DSizes<int, 2> rest_by_one(rest_size, 1); in operator()
45 rest_by_one.set(0, rest_size); in operator()
86 const int rest_size = input.size() / depth; in operator() local
90 Eigen::DSizes<Index, 2> rest_by_depth(rest_size, depth); in operator()
92 Eigen::DSizes<Index, 2> rest_by_one(rest_size, 1); in operator()
98 rest_by_one.set(0, rest_size); in operator()
Dbias_op.h35 const int64_t rest_size = input.size() / bias_size; in operator() local
37 Eigen::DSizes<int64_t, 1> bcast(rest_size); in operator()
42 const int rest_size = input.size() / bias_size; in operator() local
44 Eigen::DSizes<int, 1> bcast(rest_size); in operator()
Dfused_batch_norm_op.h83 const int rest_size = input.size() / depth; in operator() local
85 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator()
89 Eigen::array<int, 2> rest_by_one({rest_size, 1}); in operator()
95 rest_by_one.set(0, rest_size); in operator()
Dfused_batch_norm_op.cc94 const int rest_size = size / depth; in operator ()() local
95 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()()
100 Eigen::array<int, 2> bcast_spec({rest_size, 1}); in operator ()()
106 bcast_spec.set(0, rest_size); in operator ()()
110 const int rest_size_minus_one = (rest_size > 1) ? (rest_size - 1) : 1; in operator ()()
111 U rest_size_inv = static_cast<U>(1.0f / static_cast<U>(rest_size)); in operator ()()
114 static_cast<U>(rest_size) / static_cast<U>(rest_size_minus_one); in operator ()()
186 const int rest_size = size / depth; in operator ()() local
187 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()()
192 Eigen::array<int, 2> bcast_spec({rest_size, 1}); in operator ()()
[all …]