Searched refs:task_2d (Results 1 – 8 of 8) 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()285 … argmax_pooling_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_argmax_pooling_multipass; in xnn_setup_argmax_pooling2d_nhwc_f32()
597 …average_pooling_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_pixelwise_average_poolin… in setup_average_pooling2d()600 …average_pooling_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_pixelwise_average_poolin… in setup_average_pooling2d()625 … average_pooling_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_average_pooling_unipass; in setup_average_pooling2d()628 …average_pooling_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_average_pooling_multipas… in setup_average_pooling2d()
224 unpooling_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_unpooling; in xnn_setup_unpooling2d_nhwc_x32()
316 max_pooling_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_max_pooling; in setup_max_pooling2d_nhwc()
710 convolution_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_dwconv2d_chw; in setup_convolution2d_nchw()
1134 convolution_op->compute.task_2d = (pthreadpool_task_2d_t) xnn_compute_dwconv_unipass; in setup_convolution2d_nhwc()
44 pthreadpool_task_2d_t task_2d; member
1102 op->compute.task_2d, in xnn_run_operator()