Searched refs:argmax_pooling (Results 1 – 2 of 2) sorted by relevance
259 argmax_pooling_op->context.argmax_pooling = (struct argmax_pooling_context) { in xnn_setup_argmax_pooling2d_nhwc_f32()281 argmax_pooling_op->context.argmax_pooling.unipass_ukernel = ukernel->up; in xnn_setup_argmax_pooling2d_nhwc_f32()284 argmax_pooling_op->context.argmax_pooling.multipass_ukernel = ukernel->mp; in xnn_setup_argmax_pooling2d_nhwc_f32()
308 struct argmax_pooling_context argmax_pooling; member