Searched refs:task_1d (Results 1 – 9 of 9) sorted by relevance
175 … channel_shuffle_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_channel_shuffle_fixed; in setup_channel_shuffle_nc()179 … channel_shuffle_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_channel_shuffle_fixed; in setup_channel_shuffle_nc()183 … channel_shuffle_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_channel_shuffle_fixed; in setup_channel_shuffle_nc()187 …channel_shuffle_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_channel_shuffle_variable; in setup_channel_shuffle_nc()
178 softmax_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_u8_softmax; in xnn_setup_softmax_nc_qu8()294 softmax_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_f32_three_pass_softmax; in xnn_setup_softmax_nc_f32()
146 depth_to_space_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_depthtospace2d_chw2hwc; in xnn_setup_depth_to_space_nchw2nhwc_x32()
189 …global_average_pooling_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_global_average_po… in setup_global_average_pooling_nwc()192 …global_average_pooling_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_global_average_po… in setup_global_average_pooling_nwc()
213 sigmoid_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_lut_strided; in xnn_setup_sigmoid_nc_qu8()
219 leaky_relu_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_lut_strided; in xnn_setup_leaky_relu_nc_qu8()
500 …average_pooling_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_global_average_pooling_n… in setup_average_pooling2d()503 …average_pooling_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_global_average_pooling_n… in setup_average_pooling2d()
42 pthreadpool_task_1d_t task_1d; member
1081 op->compute.task_1d, in xnn_run_operator()