Searched refs:input_stride_in_bytes (Results 1 – 2 of 2) sorted by relevance
172 …const size_t input_stride_in_bytes = global_average_pooling_op->input_pixel_stride << log2_element… in setup_global_average_pooling_nwc() local177 .input_pixel_stride = input_stride_in_bytes, in setup_global_average_pooling_nwc()178 .input_batch_stride = input_stride_in_bytes * width, in setup_global_average_pooling_nwc()
483 …const size_t input_stride_in_bytes = average_pooling_op->input_pixel_stride << log2_input_element_… in setup_average_pooling2d() local488 .input_pixel_stride = input_stride_in_bytes, in setup_average_pooling2d()489 .input_batch_stride = input_stride_in_bytes * input_elements, in setup_average_pooling2d()