| /external/tensorflow/tensorflow/core/kernels/ |
| bias_op.h | 35 const int64_t rest_size = input.size() / bias_size; local
37 Eigen::DSizes<int64_t, 1> bcast(rest_size);
42 const int rest_size = input.size() / bias_size; local
44 Eigen::DSizes<int, 1> bcast(rest_size);
|
| batch_norm_op.h | 36 const int rest_size = input.size() / depth; local
38 Eigen::DSizes<int, 2> rest_by_depth(rest_size, depth);
40 Eigen::DSizes<int, 2> rest_by_one(rest_size, 1);
45 rest_by_one.set(0, rest_size);
86 const int rest_size = input.size() / depth; local
90 Eigen::DSizes<Index, 2> rest_by_depth(rest_size, depth);
92 Eigen::DSizes<Index, 2> rest_by_one(rest_size, 1);
98 rest_by_one.set(0, rest_size);
|
| fused_batch_norm_op.h | 83 const int rest_size = input.size() / depth; local
85 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth);
89 Eigen::array<int, 2> rest_by_one({rest_size, 1});
95 rest_by_one.set(0, rest_size);
|
| fused_batch_norm_op.cc | 77 const int rest_size = size / depth; local
78 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth);
83 Eigen::array<int, 2> bcast_spec({rest_size, 1});
89 bcast_spec.set(0, rest_size);
93 const int rest_size_minus_one = (rest_size > 1) ? (rest_size - 1) : 1;
94 U rest_size_inv = static_cast<U>(1.0f / static_cast<U>(rest_size));
97 static_cast<U>(rest_size) / static_cast<U>(rest_size_minus_one);
165 const int rest_size = size / depth; local
166 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth)
[all...] |
