Lines Matching refs:leaky_relu_op

35   xnn_operator_t leaky_relu_op = NULL;  in xnn_create_leaky_relu_nc_qu8()  local
117 leaky_relu_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_leaky_relu_nc_qu8()
118 if (leaky_relu_op == NULL) { in xnn_create_leaky_relu_nc_qu8()
125 leaky_relu_op->lookup_table = xnn_allocate_simd_memory(256 * sizeof(uint8_t)); in xnn_create_leaky_relu_nc_qu8()
126 if (leaky_relu_op->lookup_table == NULL) { in xnn_create_leaky_relu_nc_qu8()
133 uint8_t* lookup_table = leaky_relu_op->lookup_table; in xnn_create_leaky_relu_nc_qu8()
148 leaky_relu_op->channels = channels; in xnn_create_leaky_relu_nc_qu8()
149 leaky_relu_op->input_pixel_stride = input_stride; in xnn_create_leaky_relu_nc_qu8()
150 leaky_relu_op->output_pixel_stride = output_stride; in xnn_create_leaky_relu_nc_qu8()
152 leaky_relu_op->type = xnn_operator_type_leaky_relu_nc_qu8; in xnn_create_leaky_relu_nc_qu8()
154 leaky_relu_op->state = xnn_run_state_invalid; in xnn_create_leaky_relu_nc_qu8()
156 *leaky_relu_op_out = leaky_relu_op; in xnn_create_leaky_relu_nc_qu8()
160 xnn_delete_operator(leaky_relu_op); in xnn_create_leaky_relu_nc_qu8()
165 xnn_operator_t leaky_relu_op, in xnn_setup_leaky_relu_nc_qu8() argument
171 if (leaky_relu_op->type != xnn_operator_type_leaky_relu_nc_qu8) { in xnn_setup_leaky_relu_nc_qu8()
174 xnn_operator_type_to_string(leaky_relu_op->type)); in xnn_setup_leaky_relu_nc_qu8()
177 leaky_relu_op->state = xnn_run_state_invalid; in xnn_setup_leaky_relu_nc_qu8()
187 leaky_relu_op->state = xnn_run_state_skip; in xnn_setup_leaky_relu_nc_qu8()
191 const size_t channels = leaky_relu_op->channels; in xnn_setup_leaky_relu_nc_qu8()
192 const size_t input_stride = leaky_relu_op->input_pixel_stride; in xnn_setup_leaky_relu_nc_qu8()
193 const size_t output_stride = leaky_relu_op->output_pixel_stride; in xnn_setup_leaky_relu_nc_qu8()
196 leaky_relu_op->context.lut_contiguous = (struct lut_contiguous_context) { in xnn_setup_leaky_relu_nc_qu8()
199 .t = leaky_relu_op->lookup_table, in xnn_setup_leaky_relu_nc_qu8()
204 leaky_relu_op->compute.type = xnn_parallelization_type_1d_tile_1d; in xnn_setup_leaky_relu_nc_qu8()
205leaky_relu_op->compute.task_1d_tile_1d = (pthreadpool_task_1d_tile_1d_t) xnn_compute_lut_contiguou… in xnn_setup_leaky_relu_nc_qu8()
206 leaky_relu_op->compute.range[0] = batch_size * channels * sizeof(uint8_t); in xnn_setup_leaky_relu_nc_qu8()
207 leaky_relu_op->compute.tile[0] = block_size; in xnn_setup_leaky_relu_nc_qu8()
209 leaky_relu_op->context.lut_strided = (struct lut_strided_context) { in xnn_setup_leaky_relu_nc_qu8()
213 .t = leaky_relu_op->lookup_table, in xnn_setup_leaky_relu_nc_qu8()
218 leaky_relu_op->compute.type = xnn_parallelization_type_1d; in xnn_setup_leaky_relu_nc_qu8()
219 leaky_relu_op->compute.task_1d = (pthreadpool_task_1d_t) xnn_compute_lut_strided; in xnn_setup_leaky_relu_nc_qu8()
220 leaky_relu_op->compute.range[0] = batch_size; in xnn_setup_leaky_relu_nc_qu8()
221 leaky_relu_op->compute.tile[0] = 0; in xnn_setup_leaky_relu_nc_qu8()
223 leaky_relu_op->state = xnn_run_state_ready; in xnn_setup_leaky_relu_nc_qu8()