Home
last modified time | relevance | path

Searched refs:leaky_relu_op (Results 1 – 4 of 4) sorted by relevance

/external/XNNPACK/src/operators/
Dleaky-relu-nc.c35 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()
[all …]
Dunary-elementwise-nc.c689 xnn_operator_t leaky_relu_op, in xnn_setup_leaky_relu_nc_f32() argument
695 if (leaky_relu_op->type != xnn_operator_type_leaky_relu_nc_f32) { in xnn_setup_leaky_relu_nc_f32()
698 xnn_operator_type_to_string(leaky_relu_op->type)); in xnn_setup_leaky_relu_nc_f32()
701 leaky_relu_op->state = xnn_run_state_invalid; in xnn_setup_leaky_relu_nc_f32()
704 leaky_relu_op, in xnn_setup_leaky_relu_nc_f32()
707 &leaky_relu_op->params.f32_lrelu, sizeof(leaky_relu_op->params.f32_lrelu)); in xnn_setup_leaky_relu_nc_f32()
/external/XNNPACK/test/
Dleaky-relu-operator-tester.h179 xnn_operator_t leaky_relu_op = nullptr; in TestF32() local
185 0, &leaky_relu_op)); in TestF32()
186 ASSERT_NE(nullptr, leaky_relu_op); in TestF32()
189 …_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_leaky_relu_op(leaky_relu_op, xnn_delete_op… in TestF32()
193 leaky_relu_op, in TestF32()
199 xnn_run_operator(leaky_relu_op, nullptr /* thread pool */)); in TestF32()
237 xnn_operator_t leaky_relu_op = nullptr; in TestQU8() local
246 0, &leaky_relu_op)); in TestQU8()
247 ASSERT_NE(nullptr, leaky_relu_op); in TestQU8()
250 …_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_leaky_relu_op(leaky_relu_op, xnn_delete_op… in TestQU8()
[all …]
/external/XNNPACK/include/
Dxnnpack.h1397 xnn_operator_t leaky_relu_op,
2160 xnn_operator_t leaky_relu_op,