Searched refs:xnn_compute_argmax_pooling_unipass (Results 1 – 3 of 3) sorted by relevance
282 argmax_pooling_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_argmax_pooling_unipass; in xnn_setup_argmax_pooling2d_nhwc_f32()
621 XNN_PRIVATE void xnn_compute_argmax_pooling_unipass(
675 void xnn_compute_argmax_pooling_unipass( in xnn_compute_argmax_pooling_unipass() function