Searched refs:xnn_compute_argmax_pooling_multipass (Results 1 – 3 of 3) sorted by relevance
285 … argmax_pooling_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_argmax_pooling_multipass; in xnn_setup_argmax_pooling2d_nhwc_f32()
626 XNN_PRIVATE void xnn_compute_argmax_pooling_multipass(
694 void xnn_compute_argmax_pooling_multipass( in xnn_compute_argmax_pooling_multipass() function