| /OK3568_Linux_fs/external/rknpu2/examples/rknn_matmul_api_demo/src/ |
| H A D | rknn_matmul_demo.cpp | 35 if (!attr->n_dims) { in get_dims_string()
41 for (uint32_t i = 1; i < attr->n_dims; ++i) { in get_dims_string()
59 if (attr->n_dims == 2) { in dump_matmul_tensor()
74 else if (attr->n_dims == 3) { in dump_matmul_tensor()
91 else if (attr->n_dims == 4) { in dump_matmul_tensor()
193 if (io_attr.B.n_dims == 2) { in main()
201 else if (io_attr.B.n_dims == 4) { in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/ |
| H A D | rknn_dynshape_inference.cc | 74 std::string shape_str = attr->n_dims < 1 ? "" : std::to_string(attr->dims[0]); in dump_tensor_attr()
75 for (int i = 1; i < attr->n_dims; ++i) in dump_tensor_attr()
83 …attr->index, attr->name, attr->n_dims, shape_str.c_str(), attr->n_elems, attr->size, attr->w_strid… in dump_tensor_attr()
94 range_str += dyn_range->n_dims < 1 ? "" : std::to_string(dyn_range->dyn_range[n][0]); in dump_input_dynamic_range()
95 for (int i = 1; i < dyn_range->n_dims; ++i) in dump_input_dynamic_range()
173 for (uint32_t i = 0; i < output_attr->n_dims; ++i) in save_npy()
466 for (int j = 0; j < input_attrs[i].n_dims; ++j) in main()
|
| H A D | rknn_dynshape_inference_zero_copy.cc | 74 std::string shape_str = attr->n_dims < 1 ? "" : std::to_string(attr->dims[0]); in dump_tensor_attr()
75 for (int i = 1; i < attr->n_dims; ++i) in dump_tensor_attr()
83 …attr->index, attr->name, attr->n_dims, shape_str.c_str(), attr->n_elems, attr->size, attr->w_strid… in dump_tensor_attr()
94 range_str += dyn_range->n_dims < 1 ? "" : std::to_string(dyn_range->dyn_range[n][0]); in dump_input_dynamic_range()
95 for (int i = 1; i < dyn_range->n_dims; ++i) in dump_input_dynamic_range()
173 for (uint32_t i = 0; i < output_attr->n_dims; ++i) in save_npy()
489 for (int j = 0; j < input_attrs[i].n_dims; ++j) in main()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/ |
| H A D | main.cc | 49 for (int i = 0; i < attr->n_dims; ++i) in dump_tensor_attr()
52 sprintf(&dims[idx], "%d%s", attr->dims[i], (i == attr->n_dims - 1) ? "" : ", "); in dump_tensor_attr()
56 …attr->index, attr->name, attr->n_dims, dims, attr->n_elems, attr->size, get_format_string(attr->fm… in dump_tensor_attr()
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()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_mobilenet_demo/src/ |
| H A D | main.cc | 111 for (int i = 0; i < attr->n_dims; ++i) in dump_tensor_attr()
114 sprintf(&dims[idx], "%d%s", attr->dims[i], (i == attr->n_dims - 1) ? "" : ", "); in dump_tensor_attr()
118 …attr->index, attr->name, attr->n_dims, dims, attr->n_elems, attr->size, get_format_string(attr->fm… in dump_tensor_attr()
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()
|
| H A D | main_nhwc.cc | 111 for (int i = 0; i < attr->n_dims; ++i) in dump_tensor_attr()
114 sprintf(&dims[idx], "%d%s", attr->dims[i], (i == attr->n_dims - 1) ? "" : ", "); in dump_tensor_attr()
118 …attr->index, attr->name, attr->n_dims, dims, attr->n_elems, attr->size, get_format_string(attr->fm… in dump_tensor_attr()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/ |
| H A D | rknn_internal_mem_reuse_demo.cc | 81 for (int i = 0; i < attr->n_dims; ++i) { in dump_tensor_attr()
83 sprintf(&dims[idx], "%d%s", attr->dims[i], (i == attr->n_dims - 1) ? "" : ", "); in dump_tensor_attr()
87 …attr->index, attr->name, attr->n_dims, dims, attr->n_elems, attr->size, get_format_string(attr->fm… in dump_tensor_attr()
288 for (uint32_t i = 0; i < output_attr->n_dims; ++i) { in save_npy()
297 …if (cnpy_create(output_path, byte_order, dtype, order, output_attr->n_dims, (const size_t*)output_… in save_npy()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/ |
| H A D | rknn_benchmark.cpp | 74 std::string shape_str = attr->n_dims < 1 ? "" : std::to_string(attr->dims[0]); in dump_tensor_attr()
75 for (int i = 1; i < attr->n_dims; ++i) { in dump_tensor_attr()
82 …attr->index, attr->name, attr->n_dims, shape_str.c_str(), attr->n_elems, attr->size, attr->w_strid… in dump_tensor_attr()
170 for (uint32_t i = 0; i < output_attr->n_dims; ++i) { in save_npy()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/ |
| H A D | main.cc | 42 std::string shape_str = attr->n_dims < 1 ? "" : std::to_string(attr->dims[0]); in dump_tensor_attr()
43 for (int i = 1; i < attr->n_dims; ++i) { in dump_tensor_attr()
50 …attr->index, attr->name, attr->n_dims, shape_str.c_str(), attr->n_elems, attr->size, attr->w_strid… in dump_tensor_attr()
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RK3588/Android/librknn_api/include/ |
| H A D | rknn_api.h | 239 uint32_t n_dims; /* the number of dimensions. */ member
274 uint32_t n_dims; /* the number of dimensions. */ member
|
| H A D | rknn_matmul_api.h | 29 uint32_t n_dims; member
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RK356X/Linux/librknn_api/include/ |
| H A D | rknn_api.h | 239 uint32_t n_dims; /* the number of dimensions. */ member
274 uint32_t n_dims; /* the number of dimensions. */ member
|
| H A D | rknn_matmul_api.h | 29 uint32_t n_dims; member
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RK3588/Linux/librknn_api/include/ |
| H A D | rknn_api.h | 239 uint32_t n_dims; /* the number of dimensions. */ member
274 uint32_t n_dims; /* the number of dimensions. */ member
|
| H A D | rknn_matmul_api.h | 29 uint32_t n_dims; member
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RK356X/Android/librknn_api/include/ |
| H A D | rknn_api.h | 239 uint32_t n_dims; /* the number of dimensions. */ member
274 uint32_t n_dims; /* the number of dimensions. */ member
|
| H A D | rknn_matmul_api.h | 29 uint32_t n_dims; member
|
| /OK3568_Linux_fs/external/rknpu2/runtime/RV1106/Linux/librknn_api/include/ |
| H A D | rknn_api.h | 239 uint32_t n_dims; /* the number of dimensions. */ member
274 uint32_t n_dims; /* the number of dimensions. */ member
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/ |
| H A D | rknn_api.h | 239 uint32_t n_dims; /* the number of dimensions. */ member
274 uint32_t n_dims; /* the number of dimensions. */ member
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_mobilenet_demo/src/ |
| H A D | main.cc | 42 … attr->index, attr->name, attr->n_dims, attr->dims[0], attr->dims[1], attr->dims[2], attr->dims[3], in dump_tensor_attr()
|
| /OK3568_Linux_fs/external/rknpu2/examples/librknn_api_android_demo/include/ |
| H A D | rknn_api.h | 232 uint32_t n_dims; /* the number of dimensions. */ member
|
| /OK3568_Linux_fs/external/rknpu2/doc/ |
| H A D | RKNN_Dynamic_Shape_Usage.md | 58 for (int j = 0; j < input_attrs[i].n_dims; ++j)
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_multiple_input_demo/src/ |
| H A D | main.cc | 64 … attr->index, attr->name, attr->n_dims, attr->dims[0], attr->dims[1], attr->dims[2], attr->dims[3], in dump_tensor_attr()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_common_test/src/ |
| H A D | main.cc | 73 … attr->index, attr->name, attr->n_dims, attr->dims[0], attr->dims[1], attr->dims[2], attr->dims[3], in dump_tensor_attr()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_api_demo/src/ |
| H A D | rknn_create_mem_demo.cpp | 72 … attr->index, attr->name, attr->n_dims, attr->dims[0], attr->dims[1], attr->dims[2], attr->dims[3], in dump_tensor_attr()
|