Searched refs:orig_output_attrs (Results 1 – 2 of 2) sorted by relevance
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/ |
| H A D | main.cc | 440 rknn_tensor_attr orig_output_attrs[io_num.n_output]; in main() local
441 memset(orig_output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
444 orig_output_attrs[i].index = i; in main()
446 … ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(orig_output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
452 dump_tensor_attr(&orig_output_attrs[i]); in main()
458 int size = orig_output_attrs[i].size_with_stride * sizeof(float); in main()
466 int channel = orig_output_attrs[i].dims[1]; in main()
467 int h = orig_output_attrs[i].n_dims > 2 ? orig_output_attrs[i].dims[2] : 1; in main()
468 int w = orig_output_attrs[i].n_dims > 3 ? orig_output_attrs[i].dims[3] : 1; in main()
492 uint32_t sz = orig_output_attrs[i].n_elems; in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/ |
| H A D | main.cc | 412 rknn_tensor_attr orig_output_attrs[io_num.n_output]; in main() local
413 memset(orig_output_attrs, 0, io_num.n_output * sizeof(rknn_tensor_attr)); in main()
416 orig_output_attrs[i].index = i; in main()
418 … ret = rknn_query(ctx, RKNN_QUERY_OUTPUT_ATTR, &(orig_output_attrs[i]), sizeof(rknn_tensor_attr)); in main()
424 dump_tensor_attr(&orig_output_attrs[i]); in main()
430 int size = orig_output_attrs[i].size_with_stride; in main()
436 int channel = orig_output_attrs[i].dims[1]; in main()
437 int h = orig_output_attrs[i].n_dims > 2 ? orig_output_attrs[i].dims[2] : 1; in main()
438 int w = orig_output_attrs[i].n_dims > 3 ? orig_output_attrs[i].dims[3] : 1; in main()
|