Home
last modified time | relevance | path

Searched refs:input_channel_ptr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dpooling.h93 const int8* input_channel_ptr = input_row_ptr; in MaxPool() local
98 int8x16_t input_reg = vld1q_s8(input_channel_ptr); in MaxPool()
99 input_channel_ptr += 16; in MaxPool()
106 int8x8_t input_reg = vld1_s8(input_channel_ptr); in MaxPool()
107 input_channel_ptr += 8; in MaxPool()
113 acc[channel] = std::max(acc[channel], *input_channel_ptr++); in MaxPool()
204 const int8* input_channel_ptr = input_row_ptr; in AveragePool() local
209 int8x16_t input_reg = vld1q_s8(input_channel_ptr); in AveragePool()
210 input_channel_ptr += 16; in AveragePool()
223 int16x8_t input_reg = vmovl_s8(vld1_s8(input_channel_ptr)); in AveragePool()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3293 const uint8* input_channel_ptr = input_row_ptr; in AveragePool() local
3298 uint8x16_t input_reg = vld1q_u8(input_channel_ptr); in AveragePool()
3299 input_channel_ptr += 16; in AveragePool()
3312 uint16x8_t input_reg = vmovl_u8(vld1_u8(input_channel_ptr)); in AveragePool()
3313 input_channel_ptr += 8; in AveragePool()
3324 acc[channel] += *input_channel_ptr++; in AveragePool()
3481 const uint8* input_channel_ptr = input_row_ptr; in MaxPool() local
3486 uint8x16_t input_reg = vld1q_u8(input_channel_ptr); in MaxPool()
3487 input_channel_ptr += 16; in MaxPool()
3494 uint8x8_t input_reg = vld1_u8(input_channel_ptr); in MaxPool()
[all …]