Searched refs:npy_shape (Results 1 – 4 of 4) sorted by relevance
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/ |
| H A D | rknn_benchmark.cpp | 135 int npy_shape[4] = {1, 1, 1, 1}; in load_npy() local
139 npy_shape[start + i] = npy_data.shape[i]; in load_npy()
142 int height = npy_shape[1]; in load_npy()
143 int width = npy_shape[2]; in load_npy()
144 int channel = npy_shape[3]; in load_npy()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/ |
| H A D | rknn_internal_mem_reuse_demo.cc | 253 int npy_shape[4] = {1, 1, 1, 1}; in load_npy() local
257 npy_shape[start + i] = npy_data.dims[i]; in load_npy()
260 int height = npy_shape[1]; in load_npy()
261 int width = npy_shape[2]; in load_npy()
262 int channel = npy_shape[3]; in load_npy()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/ |
| H A D | rknn_dynshape_inference.cc | 134 int npy_shape[4] = {1, 1, 1, 1}; in load_npy() local
139 npy_shape[start + i] = npy_data.shape[i]; in load_npy()
142 int height = npy_shape[1]; in load_npy()
143 int width = npy_shape[2]; in load_npy()
144 int channel = npy_shape[3]; in load_npy()
|
| H A D | rknn_dynshape_inference_zero_copy.cc | 134 int npy_shape[4] = {1, 1, 1, 1}; in load_npy() local
139 npy_shape[start + i] = npy_data.shape[i]; in load_npy()
142 int height = npy_shape[1]; in load_npy()
143 int width = npy_shape[2]; in load_npy()
144 int channel = npy_shape[3]; in load_npy()
|