Lines Matching refs:unary_elementwise_op

31   xnn_operator_t unary_elementwise_op = NULL;  in create_unary_elementwise_nc()  local
62 unary_elementwise_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in create_unary_elementwise_nc()
63 if (unary_elementwise_op == NULL) { in create_unary_elementwise_nc()
70 unary_elementwise_op->channels = channels; in create_unary_elementwise_nc()
71 unary_elementwise_op->input_pixel_stride = input_stride; in create_unary_elementwise_nc()
72 unary_elementwise_op->output_pixel_stride = output_stride; in create_unary_elementwise_nc()
74 memcpy(&unary_elementwise_op->params, params, params_size); in create_unary_elementwise_nc()
77 unary_elementwise_op->ukernel.vunary.function = ukernel; in create_unary_elementwise_nc()
78 unary_elementwise_op->type = operator_type; in create_unary_elementwise_nc()
80 unary_elementwise_op->state = xnn_run_state_invalid; in create_unary_elementwise_nc()
82 *unary_elementwise_op_out = unary_elementwise_op; in create_unary_elementwise_nc()
87 xnn_operator_t unary_elementwise_op, in setup_unary_elementwise_nc() argument
97 xnn_operator_type_to_string(unary_elementwise_op->type)); in setup_unary_elementwise_nc()
102 unary_elementwise_op->state = xnn_run_state_skip; in setup_unary_elementwise_nc()
106 const size_t channels = unary_elementwise_op->channels; in setup_unary_elementwise_nc()
107 const size_t input_stride = unary_elementwise_op->input_pixel_stride; in setup_unary_elementwise_nc()
108 const size_t output_stride = unary_elementwise_op->output_pixel_stride; in setup_unary_elementwise_nc()
110 xnn_univector_ukernel_function ukernel = unary_elementwise_op->ukernel.vunary.function; in setup_unary_elementwise_nc()
114 unary_elementwise_op->context.univector_contiguous = (struct univector_contiguous_context) { in setup_unary_elementwise_nc()
122 memcpy(&unary_elementwise_op->context.univector_contiguous.params, params, params_size); in setup_unary_elementwise_nc()
124 unary_elementwise_op->compute.type = xnn_parallelization_type_1d_tile_1d; in setup_unary_elementwise_nc()
125unary_elementwise_op->compute.task_1d_tile_1d = (pthreadpool_task_1d_tile_1d_t) xnn_compute_univec… in setup_unary_elementwise_nc()
126 unary_elementwise_op->compute.range[0] = (batch_size * channels) << log2_element_size; in setup_unary_elementwise_nc()
127 unary_elementwise_op->compute.tile[0] = block_size; in setup_unary_elementwise_nc()
129 unary_elementwise_op->context.univector_strided = (struct univector_strided_context) { in setup_unary_elementwise_nc()
138 memcpy(&unary_elementwise_op->context.univector_strided.params, params, params_size); in setup_unary_elementwise_nc()
140 unary_elementwise_op->compute.type = xnn_parallelization_type_1d_tile_1d; in setup_unary_elementwise_nc()
141unary_elementwise_op->compute.task_1d_tile_1d = (pthreadpool_task_1d_tile_1d_t) xnn_compute_univec… in setup_unary_elementwise_nc()
142 unary_elementwise_op->compute.range[0] = batch_size; in setup_unary_elementwise_nc()
143 unary_elementwise_op->compute.tile[0] = 1; in setup_unary_elementwise_nc()
145 unary_elementwise_op->state = xnn_run_state_ready; in setup_unary_elementwise_nc()