Home
last modified time | relevance | path

Searched refs:depth_base (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dpooling.h67 for (int depth_base = 0; depth_base < depth; in MaxPool() local
68 depth_base += kPoolingAccTrancheSize) { in MaxPool()
70 std::min(depth - depth_base, kPoolingAccTrancheSize); in MaxPool()
86 input_data + depth_base + in MaxPool()
119 out_x, depth_base); in MaxPool()
177 for (int depth_base = 0; depth_base < depth; in AveragePool() local
178 depth_base += kPoolingAccTrancheSize) { in AveragePool()
180 std::min(depth - depth_base, kPoolingAccTrancheSize); in AveragePool()
197 input_data + depth_base + in AveragePool()
241 out_x, depth_base); in AveragePool()
/external/mesa3d/src/gallium/drivers/freedreno/a2xx/
Dfd2_gmem.c511 uint32_t color_base = 0, depth_base = gmem->zsbuf_base[0]; in fd2_emit_tile_init() local
519 depth_base = size / 2; in fd2_emit_tile_init()
524 color_base = depth_base; in fd2_emit_tile_init()
525 depth_base = depth_base + size / 2; in fd2_emit_tile_init()
546 patch->cs[5] = A2XX_RB_DEPTH_INFO_DEPTH_BASE(depth_base) | in fd2_emit_tile_init()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3266 for (int depth_base = 0; depth_base < depth; in AveragePool() local
3267 depth_base += kPoolingAccTrancheSize) { in AveragePool()
3269 std::min(depth - depth_base, kPoolingAccTrancheSize); in AveragePool()
3286 input_data + depth_base + in AveragePool()
3330 out_x, depth_base); in AveragePool()
3456 for (int depth_base = 0; depth_base < depth; in MaxPool() local
3457 depth_base += kPoolingAccTrancheSize) { in MaxPool()
3459 std::min(depth - depth_base, kPoolingAccTrancheSize); in MaxPool()
3474 input_data + depth_base + in MaxPool()
3507 out_x, depth_base); in MaxPool()