Searched refs:argmax_pooling_context (Results 1 – 4 of 4) sorted by relevance
539 struct argmax_pooling_context { struct563 const struct argmax_pooling_context context[restrict XNN_MIN_ELEMENTS(1)], argument568 const struct argmax_pooling_context context[restrict XNN_MIN_ELEMENTS(1)],
308 struct argmax_pooling_context argmax_pooling;
259 argmax_pooling_op->context.argmax_pooling = (struct argmax_pooling_context) { in xnn_setup_argmax_pooling2d_nhwc_f32()
460 const struct argmax_pooling_context context[restrict XNN_MIN_ELEMENTS(1)], in xnn_compute_argmax_pooling_unipass()479 const struct argmax_pooling_context context[restrict XNN_MIN_ELEMENTS(1)], in xnn_compute_argmax_pooling_multipass()