Searched refs:index_height_stride (Results 1 – 4 of 4) sorted by relevance
255 const size_t index_height_stride = output_width * channels * sizeof(uint32_t); in xnn_setup_argmax_pooling2d_nhwc_f32() local269 .index_batch_stride = output_height * index_height_stride, in xnn_setup_argmax_pooling2d_nhwc_f32()270 .index_height_stride = index_height_stride, in xnn_setup_argmax_pooling2d_nhwc_f32()
213 .index_height_stride = input_width * channels * sizeof(uint32_t), in xnn_setup_unpooling2d_nhwc_x32()
521 size_t index_height_stride; member550 size_t index_height_stride; member
470 batch_index * context->index_batch_stride + output_y * context->index_height_stride); in xnn_compute_argmax_pooling_unipass()489 batch_index * context->index_batch_stride + output_y * context->index_height_stride); in xnn_compute_argmax_pooling_multipass()526 input_y * context->index_height_stride + input_x * context->index_width_stride); in xnn_compute_unpooling()