| /OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/ |
| H A D | rknn_dynshape_inference.cc | 72 static void dump_tensor_attr(rknn_tensor_attr *attr) in dump_tensor_attr()
107 static unsigned char *load_npy(const char *input_path, rknn_tensor_attr *input_attr, int *input_siz… in load_npy()
169 static void save_npy(const char *output_path, T *output_data, rknn_tensor_attr *output_attr) in save_npy()
182 static unsigned char *load_image(const char *image_path, rknn_tensor_attr *input_attr) in load_image()
357 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
358 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
363 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
373 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
374 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
379 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
[all …]
|
| H A D | rknn_dynshape_inference_zero_copy.cc | 72 static void dump_tensor_attr(rknn_tensor_attr *attr) in dump_tensor_attr()
107 static unsigned char *load_npy(const char *input_path, rknn_tensor_attr *input_attr, int *input_siz… in load_npy()
169 static void save_npy(const char *output_path, T *output_data, rknn_tensor_attr *output_attr) in save_npy()
182 static unsigned char *load_image(const char *image_path, rknn_tensor_attr *input_attr) in load_image()
357 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
358 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
363 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
373 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
374 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
379 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/doc/ |
| H A D | RKNN_Dynamic_Shape_Usage.md | 51 加载动态形状输入RKNN模型后，您可以在运行时动态修改输入的形状。通过调用rknn_set_input_shape接口，传入包含形状信息的rknn_tensor_attr指针可以设置当前次推理的形状…
72 其中，shape_num是支持的形状个数，shape_range[i]是第i个输入的rknn_input_range结构体，input_attrs[i]是第i个输入的rknn_tensor_attr…
77 rknn_tensor_attr cur_input_attrs[io_num.n_input];
78 memset(cur_input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr));
82 … = rknn_query(ctx, RKNN_QUERY_CURRENT_INPUT_ATTR, &(cur_input_attrs[i]), sizeof(rknn_tensor_attr));
91 rknn_tensor_attr cur_output_attrs[io_num.n_output];
92 memset(cur_output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr));
96 … rknn_query(ctx, RKNN_QUERY_CURRENT_OUTPUT_ATTR, &(cur_output_attrs[i]), sizeof(rknn_tensor_attr));
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/ |
| H A D | main.cc | 46 static void dump_tensor_attr(rknn_tensor_attr *attr) in dump_tensor_attr()
96 static unsigned char *load_image(const char *image_path, rknn_tensor_attr *input_attr, int *img_hei… in load_image()
280 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
281 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
286 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
296 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
297 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
302 …ret = rknn_query(ctx, RKNN_QUERY_NATIVE_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
412 rknn_tensor_attr orig_output_attrs[io_num.n_output]; in main()
413 memset(orig_output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/ |
| H A D | rknn_create_mem_demo.cpp | 68 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
77 static unsigned char* load_image(const char* image_path, rknn_tensor_attr* input_attr) in load_image()
200 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
201 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
205 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
214 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
215 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
219 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| H A D | rknn_create_mem_with_rga_demo.cpp | 74 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
83 unsigned char* load_image(const char* image_path, rknn_tensor_attr* input_attr) in load_image()
190 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
191 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
195 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
204 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
205 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
209 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| H A D | rknn_with_mmz_demo.cpp | 69 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
78 static unsigned char* load_image(const char* image_path, rknn_tensor_attr* input_attr) in load_image()
175 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
176 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
180 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
189 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
190 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
194 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| H A D | rknn_set_internal_mem_from_fd_demo.cpp | 69 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
78 static unsigned char* load_image(const char* image_path, rknn_tensor_attr* input_attr) in load_image()
184 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
185 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
189 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
198 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
199 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
203 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/ |
| H A D | rknn_create_mem_demo.cpp | 68 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
77 static unsigned char* load_image(const char* image_path, rknn_tensor_attr* input_attr) in load_image()
200 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
201 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
205 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
214 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
215 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
219 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/ |
| H A D | main.cc | 108 static void dump_tensor_attr(rknn_tensor_attr *attr) in dump_tensor_attr()
158 static unsigned char *load_image(const char *image_path, rknn_tensor_attr *input_attr) in load_image()
308 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
309 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
314 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
324 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
325 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
330 …ret = rknn_query(ctx, RKNN_QUERY_NATIVE_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
440 rknn_tensor_attr orig_output_attrs[io_num.n_output]; in main()
441 memset(orig_output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
[all …]
|
| H A D | main_nhwc.cc | 108 static void dump_tensor_attr(rknn_tensor_attr *attr) in dump_tensor_attr()
158 static unsigned char *load_image(const char *image_path, rknn_tensor_attr *input_attr) in load_image()
278 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
279 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
284 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
294 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
295 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
300 … rknn_query(ctx, RKNN_QUERY_NATIVE_NHWC_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_multiple_input_demo/src/ |
| H A D | main.cc | 60 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
122 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
123 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
127 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
136 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
137 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
141 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/ |
| H A D | main.cc | 69 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
124 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
125 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
129 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
138 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
139 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
143 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/ |
| H A D | rknn_benchmark.cpp | 72 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
87 static unsigned char* load_npy(const char* input_path, rknn_tensor_attr* input_attr, int* input_typ… in load_npy()
166 static void save_npy(const char* output_path, float* output_data, rknn_tensor_attr* output_attr) in save_npy()
178 static unsigned char* load_image(const char* image_path, rknn_tensor_attr* input_attr) in load_image()
317 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
318 memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr)); in main()
322 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
332 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
333 memset(output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
337 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/ |
| H A D | main_video.cc | 44 rknn_tensor_attr* input_attrs;
45 rknn_tensor_attr* output_attrs;
68 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
180 …rknn_tensor_attr* input_attrs = (rknn_tensor_attr*)malloc(app_ctx->io_num.n_input * sizeof(rknn_te… in init_model()
184 ret = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in init_model()
192 …rknn_tensor_attr* output_attrs = (rknn_tensor_attr*)malloc(app_ctx->io_num.n_output * sizeof(rknn_… in init_model()
196 ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in init_model()
|
| H A D | main.cc | 40 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
191 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
195 … = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
203 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
207 … = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/ |
| H A D | rknn_internal_mem_reuse_demo.cc | 78 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
155 static unsigned char* load_image(const char* image_path, rknn_tensor_attr* input_attr) in load_image()
207 static unsigned char* load_npy(const char* input_path, rknn_tensor_attr* input_attr, int* input_typ… in load_npy()
284 static int save_npy(const char* output_path, float* output_data, rknn_tensor_attr* output_attr) in save_npy()
344 rknn_tensor_attr input_attrs[TOTAL_RKNN_MODEL_NUM][1]; // this demo only support one input in main()
345 rknn_tensor_attr output_attrs[TOTAL_RKNN_MODEL_NUM][MAX_OUTPUT_NUM]; in main()
413 memset(input_attrs[n], 0, io_num[n].n_input * sizeof(rknn_tensor_attr)); in main()
417 … ret = rknn_query(ctx[n], RKNN_QUERY_INPUT_ATTR, &(input_attrs[n][i]), sizeof(rknn_tensor_attr)); in main()
426 memset(output_attrs[n], 0, io_num[n].n_output * sizeof(rknn_tensor_attr)); in main()
430 … ret = rknn_query(ctx[n], RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[n][i]), sizeof(rknn_tensor_attr)); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_mobilenet_demo/src/ |
| H A D | main.cc | 38 static void dump_tensor_attr(rknn_tensor_attr* attr) in dump_tensor_attr()
154 rknn_tensor_attr input_attrs[io_num.n_input]; in main()
158 … = rknn_query(ctx, RKNN_QUERY_INPUT_ATTR, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in main()
167 rknn_tensor_attr output_attrs[io_num.n_output]; in main()
171 … = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/ |
| H A D | yolo_image.cc | 56 rknn_tensor_attr input_attrs[1];
57 rknn_tensor_attr output_attrs[3];
124 memset(input_attrs, 0, n_input * sizeof(rknn_tensor_attr)); in create()
128 ret = rknn_query(ctx, cmd, &(input_attrs[i]), sizeof(rknn_tensor_attr)); in create()
153 memset(output_attrs, 0, n_output * sizeof(rknn_tensor_attr)); in create()
157 ret = rknn_query(ctx, cmd, &(output_attrs[i]), sizeof(rknn_tensor_attr)); in create()
|
| H A D | rknn_api.h | 265 } rknn_tensor_attr; typedef
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
691 int rknn_set_input_shape(rknn_context ctx, rknn_tensor_attr* attr);
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RK3588/Android/librknn_api/include/ |
| H A D | rknn_api.h | 265 } rknn_tensor_attr; typedef
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
691 int rknn_set_input_shape(rknn_context ctx, rknn_tensor_attr* attr);
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RK356X/Linux/librknn_api/include/ |
| H A D | rknn_api.h | 265 } rknn_tensor_attr; typedef
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
691 int rknn_set_input_shape(rknn_context ctx, rknn_tensor_attr* attr);
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RK3588/Linux/librknn_api/include/ |
| H A D | rknn_api.h | 265 } rknn_tensor_attr; typedef
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
691 int rknn_set_input_shape(rknn_context ctx, rknn_tensor_attr* attr);
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RK356X/Android/librknn_api/include/ |
| H A D | rknn_api.h | 265 } rknn_tensor_attr; typedef
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
691 int rknn_set_input_shape(rknn_context ctx, rknn_tensor_attr* attr);
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RV1106/Linux/librknn_api/include/ |
| H A D | rknn_api.h | 265 } rknn_tensor_attr; typedef
679 int rknn_set_io_mem(rknn_context ctx, rknn_tensor_mem *mem, rknn_tensor_attr *attr);
691 int rknn_set_input_shape(rknn_context ctx, rknn_tensor_attr* attr);
|