Searched refs:maxidx (Results 1 – 6 of 6) sorted by relevance
84 int maxidx = -1; in MaxPoolForwardNCHW() local91 maxidx = include_batch_in_index ? idx + offset : idx; in MaxPoolForwardNCHW()98 mask[index] = maxidx; in MaxPoolForwardNCHW()159 int maxidx = -1; in MaxPoolForwardNHWC() local166 maxidx = include_batch_in_index ? idx + offset : idx; in MaxPoolForwardNHWC()173 mask[index] = maxidx; in MaxPoolForwardNHWC()199 int maxidx = -1; in MaxPoolBackwardNoMaskNHWC() local205 maxidx = idx; in MaxPoolBackwardNoMaskNHWC()213 if (maxidx != -1) { in MaxPoolBackwardNoMaskNHWC()214 CudaAtomicAdd(bottom_diff + n * height * width * channels + maxidx, in MaxPoolBackwardNoMaskNHWC()[all …]
55 int maxidx = -1; in MaxPoolGradBackwardNoMaskNCDHW() local64 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() local115 maxidx = idx; in MaxPoolGradBackwardNoMaskNDHWC()123 if (maxidx != -1) { in MaxPoolGradBackwardNoMaskNDHWC()125 top_diff[n * plane * height * width * channels + maxidx]; in MaxPoolGradBackwardNoMaskNDHWC()
444 int maxidx = -1;453 maxidx = idx;459 if (maxidx != -1) {462 maxidx];
81 Index minidx=0, maxidx=0; in vectorVisitor() local92 maxidx = i; in vectorVisitor()100 VERIFY(maxidx == eigen_maxidx); in vectorVisitor()
72 int maxidx = 0; in FindMaxIdxWithExcludeList() local86 maxidx = i; in FindMaxIdxWithExcludeList()89 return maxidx; in FindMaxIdxWithExcludeList()
2035 size_t maxidx = nchain; in check_sysv_hash() local2045 maxidx = symsize; in check_sysv_hash()2055 else if (buf[cnt] >= maxidx) in check_sysv_hash()2065 else if (buf[cnt] >= maxidx) in check_sysv_hash()2092 size_t maxidx = nchain; in check_sysv_hash64() local2102 maxidx = symsize; in check_sysv_hash64()2112 else if (buf[cnt] >= maxidx) in check_sysv_hash64()2122 else if (buf[cnt] >= maxidx) in check_sysv_hash64()2179 size_t maxidx = shdr->sh_size / sizeof (Elf32_Word) - (4 + bitmask_words in check_gnu_hash() local2183 maxidx = MIN (maxidx, symshdr->sh_size / symshdr->sh_entsize); in check_gnu_hash()[all …]