Searched refs:rest_size (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_norm_op.h | 36 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()
|
| D | bias_op.h | 35 const Eigen::Index rest_size = input.size() / bias_size; in operator() local
37 Eigen::DSizes<Eigen::Index, 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()
|
| D | fused_batch_norm_op.cc | 160 const int rest_size = size / depth; in operator ()() local
161 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()()
166 Eigen::array<int, 2> bcast_spec({rest_size, 1}); in operator ()()
172 bcast_spec.set(0, rest_size); in operator ()()
176 const int rest_size_minus_one = (rest_size > 1) ? (rest_size - 1) : 1; in operator ()()
177 U rest_size_inv = static_cast<U>(1.0f / static_cast<U>(rest_size)); in operator ()()
180 static_cast<U>(rest_size) / static_cast<U>(rest_size_minus_one); in operator ()()
297 const int rest_size = size / depth; in operator ()() local
298 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()()
303 Eigen::array<int, 2> bcast_spec({rest_size, 1}); in operator ()()
[all …]
|
| D | fused_batch_norm_op.cu.cc | 53 const int rest_size = input.size() / depth; in operator ()() local
67 Eigen::DSizes<Eigen::Index, 2> rest_by_depth(rest_size, depth); in operator ()()
71 Eigen::array<int, 2> rest_by_one({rest_size, 1}); in operator ()()
77 rest_by_one.set(0, rest_size); in operator ()()
|
| /external/tpm2-tss/src/tss2-esys/ |
| D | esys_crypto.c | 500 size_t rest_size = data_size; in iesys_xor_parameter_obfuscation() local
510 while(rest_size > 0) { in iesys_xor_parameter_obfuscation()
518 for(size_t i = digest_size < rest_size ? digest_size : rest_size; i > 0; in iesys_xor_parameter_obfuscation()
522 rest_size = rest_size < digest_size ? 0 : rest_size - digest_size; in iesys_xor_parameter_obfuscation()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 1152 int64_t rest_size = input_type.getDimSize(0) * in matchAndRewrite() local
1157 rest_size_inv_type, {1.0f / rest_size}); in matchAndRewrite()
|