Searched refs:index_height_stride (Results 1 – 4 of 4) sorted by relevance
248 const size_t index_height_stride = output_width * channels * sizeof(uint32_t); in xnn_setup_argmax_pooling2d_nhwc_f32() local262 .index_batch_stride = output_height * index_height_stride, in xnn_setup_argmax_pooling2d_nhwc_f32()263 .index_height_stride = index_height_stride, in xnn_setup_argmax_pooling2d_nhwc_f32()
206 .index_height_stride = input_width * channels * sizeof(uint32_t), in xnn_setup_unpooling2d_nhwc_x32()
284 batch_index * context->index_batch_stride + output_y * context->index_height_stride); in xnn_compute_argmax_pooling_unipass()304 batch_index * context->index_batch_stride + output_y * context->index_height_stride); in xnn_compute_argmax_pooling_multipass()342 input_y * context->index_height_stride + input_x * context->index_width_stride); in xnn_compute_unpooling()
325 size_t index_height_stride; member354 size_t index_height_stride; member