Searched refs:unipass_ukernel (Results 1 – 6 of 6) sorted by relevance
403 xnn_dwconv_unipass_ukernel_function unipass_ukernel; member556 xnn_argmaxpool_unipass_ukernel_function unipass_ukernel; member592 xnn_avgpool_unipass_ukernel_function unipass_ukernel; member630 xnn_pavgpool_unipass_ukernel_function unipass_ukernel; member663 xnn_gavgpool_unipass_ukernel_function unipass_ukernel; member
501 average_pooling_op->context.global_average_pooling_nwc.unipass_ukernel = gavgpool->up; in setup_average_pooling2d()596 average_pooling_op->context.pixelwise_average_pooling.unipass_ukernel = pavgpool->up; in setup_average_pooling2d()624 average_pooling_op->context.average_pooling.unipass_ukernel = avgpool->up; in setup_average_pooling2d()
281 argmax_pooling_op->context.argmax_pooling.unipass_ukernel = ukernel->up; in xnn_setup_argmax_pooling2d_nhwc_f32()
190 global_average_pooling_op->context.global_average_pooling_nwc.unipass_ukernel = gavgpool->up; in setup_global_average_pooling_nwc()
1129 .unipass_ukernel = convolution_op->ukernel.dwconv.unipass_function, in setup_convolution2d_nhwc()
379 context->unipass_ukernel( in xnn_compute_dwconv_unipass()472 context->unipass_ukernel( in xnn_compute_argmax_pooling_unipass()549 context->unipass_ukernel( in xnn_compute_average_pooling_unipass()590 context->unipass_ukernel( in xnn_compute_pixelwise_average_pooling_unipass()628 context->unipass_ukernel( in xnn_compute_global_average_pooling_nwc_unipass()