Home
last modified time | relevance | path

Searched refs:output_attrs (Results 1 – 18 of 18) sorted by relevance

/OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/
H A Dyolo_image.cc57 rknn_tensor_attr output_attrs[3]; variable
153 memset(output_attrs, 0, n_output * sizeof(rknn_tensor_attr)); in create()
155 output_attrs[i].index = i; in create()
157 ret = rknn_query(ctx, cmd, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in create()
163 out_scales.push_back(output_attrs[i].scale); in create()
164 out_zps.push_back(output_attrs[i].zp); in create()
190 output_mems[i] = rknn_create_mem(ctx, output_attrs[i].n_elems * sizeof(unsigned char)); in create()
191 memset(output_mems[i]->virt_addr, 0, output_attrs[i].n_elems * sizeof(unsigned char)); in create()
193 output_attrs[i].type = RKNN_TENSOR_INT8; in create()
195 rknn_set_io_mem(ctx, output_mems[i], &(output_attrs[i])); in create()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/
H A Drknn_with_mmz_demo.cpp189 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
190 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
192 output_attrs[i].index = i; in main()
194 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
199 dump_tensor_attr(&output_attrs[i]); in main()
250 output_attrs[i].type = RKNN_TENSOR_FLOAT32; in main()
251 int output_size = output_attrs[i].n_elems * sizeof(float); in main()
252 output_attrs[i].size = output_size; in main()
305 …output_mems[i] = rknn_create_mem_from_phys(ctx, output_physs[i], output_virts[i], output_attrs[i].… in main()
318 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
[all …]
H A Drknn_create_mem_demo.cpp214 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
215 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
217 output_attrs[i].index = i; in main()
219 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
224 dump_tensor_attr(&output_attrs[i]); in main()
284 int output_size = output_attrs[i].n_elems * sizeof(float); in main()
298 output_attrs[i].type = RKNN_TENSOR_FLOAT32; in main()
300 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
326 uint32_t sz = output_attrs[i].n_elems; in main()
H A Drknn_create_mem_with_rga_demo.cpp204 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
205 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
207 output_attrs[i].index = i; in main()
209 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
214 dump_tensor_attr(&output_attrs[i]); in main()
288 int output_size = output_attrs[i].n_elems * sizeof(float); in main()
290 output_attrs[i].type = RKNN_TENSOR_FLOAT32; in main()
303 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
329 uint32_t sz = output_attrs[i].n_elems; in main()
H A Drknn_set_internal_mem_from_fd_demo.cpp198 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
199 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
201 output_attrs[i].index = i; in main()
203 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
208 dump_tensor_attr(&output_attrs[i]); in main()
311 output_attrs[i].type = RKNN_TENSOR_FLOAT32; in main()
312 int output_size = output_attrs[i].n_elems * sizeof(float); in main()
313 output_attrs[i].size = output_size; in main()
366 …output_mems[i] = rknn_create_mem_from_fd(ctx, output_fds[i], output_virts[i], output_attrs[i].size… in main()
379 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
[all …]
/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/
H A Dmain.cc324 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
325 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
328 output_attrs[i].index = i; in main()
330 …ret = rknn_query(ctx, RKNN_QUERY_NATIVE_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
336 dump_tensor_attr(&output_attrs[i]); in main()
401 output_mems[i] = rknn_create_mem(ctx, output_attrs[i].size_with_stride); in main()
416 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
464 if (output_attrs[i].fmt == RKNN_TENSOR_NC1HWC2) in main()
469 int zp = output_attrs[i].zp; in main()
470 float scale = output_attrs[i].scale; in main()
[all …]
H A Dmain_nhwc.cc294 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
295 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
298 output_attrs[i].index = i; in main()
300 …ret = rknn_query(ctx, RKNN_QUERY_NATIVE_NHWC_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_a… in main()
306 dump_tensor_attr(&output_attrs[i]); in main()
371 output_mems[i] = rknn_create_mem(ctx, output_attrs[i].n_elems * sizeof(float)); in main()
386 output_attrs[i].type = RKNN_TENSOR_FLOAT32; in main()
387 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
417 uint32_t sz = output_attrs[i].n_elems; in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/
H A Dmain.cc138 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
139 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
141 output_attrs[i].index = i; in main()
143 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
148 dump_tensor_attr(&output_attrs[i]); in main()
246 int output_size = output_attrs[i].n_elems * sizeof(float); in main()
260 output_attrs[i].type = RKNN_TENSOR_FLOAT32; in main()
262 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
288 uint32_t sz = output_attrs[i].n_elems; in main()
/OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/
H A Drknn_create_mem_demo.cpp214 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
215 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
217 output_attrs[i].index = i; in main()
219 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
224 dump_tensor_attr(&output_attrs[i]); in main()
284 int output_size = output_attrs[i].n_elems * sizeof(float); in main()
298 output_attrs[i].type = RKNN_TENSOR_FLOAT32; in main()
300 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
326 uint32_t sz = output_attrs[i].n_elems; in main()
/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/
H A Dmain.cc296 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
297 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
300 output_attrs[i].index = i; in main()
302 …ret = rknn_query(ctx, RKNN_QUERY_NATIVE_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
308 dump_tensor_attr(&output_attrs[i]); in main()
373 output_mems[i] = rknn_create_mem(ctx, output_attrs[i].size_with_stride); in main()
388 ret = rknn_set_io_mem(ctx, output_mems[i], &output_attrs[i]); in main()
440 …t8((int8_t *)output_mems[i]->virt_addr, (int8_t *)output_mems_nchw[i], (int *)output_attrs[i].dims, in main()
467 out_scales.push_back(output_attrs[i].scale); in main()
468 out_zps.push_back(output_attrs[i].zp); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/
H A Dmain_video.cc45 rknn_tensor_attr* output_attrs; member
192 …rknn_tensor_attr* output_attrs = (rknn_tensor_attr*)malloc(app_ctx->io_num.n_output * sizeof(rknn_… in init_model() local
193 memset(output_attrs, 0, sizeof(output_attrs)); in init_model()
195 output_attrs[i].index = i; in init_model()
196 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in init_model()
201 dump_tensor_attr(&(output_attrs[i])); in init_model()
205 app_ctx->output_attrs = output_attrs; in init_model()
229 free(app_ctx->output_attrs); in release_model()
303 out_scales.push_back(app_ctx->output_attrs[i].scale); in inference_model()
304 out_zps.push_back(app_ctx->output_attrs[i].zp); in inference_model()
H A Dmain.cc203 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
204 memset(output_attrs, 0, sizeof(output_attrs)); in main()
206 output_attrs[i].index = i; in main()
207 …ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_te… in main()
208 dump_tensor_attr(&(output_attrs[i])); in main()
284 out_scales.push_back(output_attrs[i].scale); in main()
285 out_zps.push_back(output_attrs[i].zp); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_mobilenet_demo/src/
H A Dmain.cc167 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
168 memset(output_attrs, 0, sizeof(output_attrs)); in main()
170 output_attrs[i].index = i; in main()
171 …ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_te… in main()
176 dump_tensor_attr(&(output_attrs[i])); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/
H A Drknn_internal_mem_reuse_demo.cc345 rknn_tensor_attr output_attrs[TOTAL_RKNN_MODEL_NUM][MAX_OUTPUT_NUM]; in main() local
426 memset(output_attrs[n], 0, io_num[n].n_output * sizeof(rknn_tensor_attr)); in main()
428 output_attrs[n][i].index = i; in main()
430 … ret = rknn_query(ctx[n], RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[n][i]), sizeof(rknn_tensor_attr)); in main()
435 dump_tensor_attr(&output_attrs[n][i]); in main()
497 output_mems[n][i] = rknn_create_mem(ctx[n], output_attrs[n][i].n_elems * sizeof(float)); in main()
503 output_attrs[n][i].type = RKNN_TENSOR_FLOAT32; in main()
504 output_attrs[n][i].fmt = RKNN_TENSOR_NCHW; in main()
505 ret = rknn_set_io_mem(ctx[n], output_mems[n][i], &output_attrs[n][i]); in main()
582 uint32_t sz = output_attrs[n][i].n_elems; in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_multiple_input_demo/src/
H A Dmain.cc136 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
137 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
139 output_attrs[i].index = i; in main()
141 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
146 dump_tensor_attr(&output_attrs[i]); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/
H A Drknn_benchmark.cpp332 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
333 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
335 output_attrs[i].index = i; in main()
337 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
343 dump_tensor_attr(&output_attrs[i]); in main()
464 save_npy(output_path, (float*)outputs[i].buf, &output_attrs[i]); in main()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/
H A Drknn_dynshape_inference_zero_copy.cc373 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
374 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
377 output_attrs[i].index = i; in main()
379 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
385 dump_tensor_attr(&output_attrs[i]); in main()
480 int output_size = output_attrs[i].size * sizeof(float); in main()
H A Drknn_dynshape_inference.cc373 rknn_tensor_attr output_attrs[io_num.n_output]; in main() local
374 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
377 output_attrs[i].index = i; in main()
379 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
385 dump_tensor_attr(&output_attrs[i]); in main()