Searched refs:index_batch_stride (Results 1 – 3 of 3) sorted by relevance
269 .index_batch_stride = output_height * index_height_stride, in xnn_setup_argmax_pooling2d_nhwc_f32()
549 size_t index_batch_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()