Searched refs:argmax_pooling_context (Results 1 – 4 of 4) sorted by relevance
598 struct argmax_pooling_context { struct622 const struct argmax_pooling_context context[restrict XNN_MIN_ELEMENTS(1)], argument627 const struct argmax_pooling_context context[restrict XNN_MIN_ELEMENTS(1)],
271 struct argmax_pooling_context argmax_pooling;
259 argmax_pooling_op->context.argmax_pooling = (struct argmax_pooling_context) { in xnn_setup_argmax_pooling2d_nhwc_f32()
676 const struct argmax_pooling_context context[restrict XNN_MIN_ELEMENTS(1)], in xnn_compute_argmax_pooling_unipass()695 const struct argmax_pooling_context context[restrict XNN_MIN_ELEMENTS(1)], in xnn_compute_argmax_pooling_multipass()