Home
last modified time | relevance | path

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

/external/XNNPACK/src/operators/
Dglobal-average-pooling-nwc.c38 xnn_operator_t global_average_pooling_op = NULL; in create_global_average_pooling_nwc() local
82 global_average_pooling_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in create_global_average_pooling_nwc()
83 if (global_average_pooling_op == NULL) { in create_global_average_pooling_nwc()
98 global_average_pooling_op->zero_buffer = zero_buffer; in create_global_average_pooling_nwc()
100 global_average_pooling_op->channels = channels; in create_global_average_pooling_nwc()
101 global_average_pooling_op->input_pixel_stride = input_stride; in create_global_average_pooling_nwc()
102 global_average_pooling_op->output_pixel_stride = output_stride; in create_global_average_pooling_nwc()
103 memcpy((void*) ((uintptr_t) global_average_pooling_op + params_offset), params, params_size); in create_global_average_pooling_nwc()
105 global_average_pooling_op->type = operator_type; in create_global_average_pooling_nwc()
106 global_average_pooling_op->flags = flags; in create_global_average_pooling_nwc()
[all …]
Dglobal-average-pooling-ncw.c27 xnn_operator_t global_average_pooling_op = NULL; in xnn_create_global_average_pooling_ncw_f32() local
68 global_average_pooling_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_global_average_pooling_ncw_f32()
69 if (global_average_pooling_op == NULL) { in xnn_create_global_average_pooling_ncw_f32()
76 global_average_pooling_op->channels = channels; in xnn_create_global_average_pooling_ncw_f32()
77 …xnn_init_f32_gavgpool_params(&global_average_pooling_op->params.f32_gavgpool, nanf(""), output_min… in xnn_create_global_average_pooling_ncw_f32()
79 global_average_pooling_op->type = xnn_operator_type_global_average_pooling_ncw_f32; in xnn_create_global_average_pooling_ncw_f32()
80 global_average_pooling_op->flags = flags; in xnn_create_global_average_pooling_ncw_f32()
82 global_average_pooling_op->state = xnn_run_state_invalid; in xnn_create_global_average_pooling_ncw_f32()
84 *global_average_pooling_op_out = global_average_pooling_op; in xnn_create_global_average_pooling_ncw_f32()
88 xnn_delete_operator(global_average_pooling_op); in xnn_create_global_average_pooling_ncw_f32()
[all …]
/external/XNNPACK/test/
Dglobal-average-pooling-operator-tester.h184 xnn_operator_t global_average_pooling_op = nullptr; in TestNWCxQU8() local
191 0, &global_average_pooling_op); in TestNWCxQU8()
196 ASSERT_NE(nullptr, global_average_pooling_op); in TestNWCxQU8()
199 … decltype(&xnn_delete_operator)> auto_global_average_pooling_op(global_average_pooling_op, xnn_del… in TestNWCxQU8()
203 global_average_pooling_op, in TestNWCxQU8()
209 xnn_run_operator(global_average_pooling_op, nullptr /* thread pool */)); in TestNWCxQU8()
253 xnn_operator_t global_average_pooling_op = nullptr; in TestNWCxQS8() local
260 0, &global_average_pooling_op); in TestNWCxQS8()
265 ASSERT_NE(nullptr, global_average_pooling_op); in TestNWCxQS8()
268 … decltype(&xnn_delete_operator)> auto_global_average_pooling_op(global_average_pooling_op, xnn_del… in TestNWCxQS8()
[all …]
/external/XNNPACK/include/
Dxnnpack.h1462 xnn_operator_t global_average_pooling_op,
1782 xnn_operator_t global_average_pooling_op,
1978 xnn_operator_t global_average_pooling_op,
2295 xnn_operator_t global_average_pooling_op,
2562 xnn_operator_t global_average_pooling_op,