Home
last modified time | relevance | path

Searched refs:maxidx (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op_gpu.cu.cc84 int maxidx = -1; in MaxPoolForwardNCHW() local
90 maxidx = idx; in MaxPoolForwardNCHW()
97 mask[index] = maxidx; in MaxPoolForwardNCHW()
160 int maxidx = -1; in MaxPoolForwardNHWC() local
166 maxidx = idx; in MaxPoolForwardNHWC()
173 mask[index] = maxidx; in MaxPoolForwardNHWC()
199 int maxidx = -1; in MaxPoolBackwardNoMaskNHWC() local
205 maxidx = idx; in MaxPoolBackwardNoMaskNHWC()
213 if (maxidx != -1) { in MaxPoolBackwardNoMaskNHWC()
214 CudaAtomicAdd(bottom_diff + n * height * width * channels + maxidx, in MaxPoolBackwardNoMaskNHWC()
[all …]
Dpooling_ops_3d_gpu.cu.cc55 int maxidx = -1; in MaxPoolGradBackwardNoMaskNCDHW() local
64 maxidx = idx; in MaxPoolGradBackwardNoMaskNCDHW()
72 if (maxidx != -1) { in MaxPoolGradBackwardNoMaskNCDHW()
74 top_diff[n * channels * plane * height * width + maxidx]; in MaxPoolGradBackwardNoMaskNCDHW()
106 int maxidx = -1; in MaxPoolGradBackwardNoMaskNDHWC() local
115 maxidx = idx; in MaxPoolGradBackwardNoMaskNDHWC()
123 if (maxidx != -1) { in MaxPoolGradBackwardNoMaskNDHWC()
125 top_diff[n * plane * height * width * channels + maxidx]; in MaxPoolGradBackwardNoMaskNDHWC()
Dpooling_ops_3d_sycl.h444 int maxidx = -1;
453 maxidx = idx;
459 if (maxidx != -1) {
462 maxidx];
/external/eigen/test/
Dvisitor.cpp81 Index minidx=0, maxidx=0; in vectorVisitor() local
92 maxidx = i; in vectorVisitor()
100 VERIFY(maxidx == eigen_maxidx); in vectorVisitor()
/external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/
Dgraph_functions_wrapper.c72 int maxidx = 0; in FindMaxIdxWithExcludeList() local
86 maxidx = i; in FindMaxIdxWithExcludeList()
89 return maxidx; in FindMaxIdxWithExcludeList()
/external/elfutils/src/
Delflint.c2014 size_t maxidx = nchain; in check_sysv_hash() local
2024 maxidx = symsize; in check_sysv_hash()
2034 else if (buf[cnt] >= maxidx) in check_sysv_hash()
2044 else if (buf[cnt] >= maxidx) in check_sysv_hash()
2065 size_t maxidx = nchain; in check_sysv_hash64() local
2075 maxidx = symsize; in check_sysv_hash64()
2085 else if (buf[cnt] >= maxidx) in check_sysv_hash64()
2095 else if (buf[cnt] >= maxidx) in check_sysv_hash64()
2152 size_t maxidx = shdr->sh_size / sizeof (Elf32_Word) - (4 + bitmask_words in check_gnu_hash() local
2156 maxidx = MIN (maxidx, symshdr->sh_size / symshdr->sh_entsize); in check_gnu_hash()
[all …]