Home
last modified time | relevance | path

Searched refs:multipass_ukernel (Results 1 – 5 of 5) sorted by relevance

/external/XNNPACK/src/xnnpack/
Dcompute.h364 xnn_argmaxpool_mp_ukernel_function multipass_ukernel; member
399 xnn_avgpool_mp_ukernel_function multipass_ukernel; member
436 xnn_pavgpool_mp_ukernel_function multipass_ukernel; member
467 xnn_gavgpool_mp_ukernel_function multipass_ukernel; member
/external/XNNPACK/src/
Dglobal-average-pooling-nwc.c302 …global_average_pooling_op->context.global_average_pooling_nwc.multipass_ukernel = xnn_params.q8.ga… in xnn_setup_global_average_pooling_nwc_q8()
366 …global_average_pooling_op->context.global_average_pooling_nwc.multipass_ukernel = xnn_params.f32.g… in xnn_setup_global_average_pooling_nwc_f32()
Daverage-pooling-nhwc.c476 average_pooling_op->context.average_pooling.multipass_ukernel = xnn_params.q8.avgpool.mp; in xnn_setup_average_pooling2d_nhwc_q8()
604 average_pooling_op->context.average_pooling.multipass_ukernel = xnn_params.f32.avgpool.mp; in xnn_setup_average_pooling2d_nhwc_f32()
662 …average_pooling_op->context.pixelwise_average_pooling.multipass_ukernel = xnn_params.f32.pavgpool.… in xnn_setup_average_pooling2d_nhwc_f32()
Doperator-run.c309 context->multipass_ukernel( in xnn_compute_argmax_pooling_multipass()
384 context->multipass_ukernel( in xnn_compute_average_pooling_multipass()
425 context->multipass_ukernel( in xnn_compute_pixelwise_average_pooling_multipass()
461 context->multipass_ukernel( in xnn_compute_global_average_pooling_nwc_multipass()
Dargmax-pooling-nhwc.c278 argmax_pooling_op->context.argmax_pooling.multipass_ukernel = ukernel->mp; in xnn_setup_argmax_pooling2d_nhwc_f32()