Searched refs:output_attr (Results 1 – 4 of 4) sorted by relevance
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_internal_mem_reuse_demo/src/ |
| H A D | rknn_internal_mem_reuse_demo.cc | 284 static int save_npy(const char* output_path, float* output_data, rknn_tensor_attr* output_attr) in save_npy() argument
288 for (uint32_t i = 0; i < output_attr->n_dims; ++i) { in save_npy()
289 size *= output_attr->dims[i]; in save_npy()
297 … (cnpy_create(output_path, byte_order, dtype, order, output_attr->n_dims, (const size_t*)output_at… in save_npy()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_benchmark/src/ |
| H A D | rknn_benchmark.cpp | 166 static void save_npy(const char* output_path, float* output_data, rknn_tensor_attr* output_attr) in save_npy() argument
170 for (uint32_t i = 0; i < output_attr->n_dims; ++i) { in save_npy()
171 output_shape.push_back(output_attr->dims[i]); in save_npy()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_dynamic_shape_input_demo/src/ |
| H A D | rknn_dynshape_inference.cc | 169 static void save_npy(const char *output_path, T *output_data, rknn_tensor_attr *output_attr) in save_npy() argument
173 for (uint32_t i = 0; i < output_attr->n_dims; ++i) in save_npy()
175 output_shape.push_back(output_attr->dims[i]); in save_npy()
|
| H A D | rknn_dynshape_inference_zero_copy.cc | 169 static void save_npy(const char *output_path, T *output_data, rknn_tensor_attr *output_attr) in save_npy() argument
173 for (uint32_t i = 0; i < output_attr->n_dims; ++i) in save_npy()
175 output_shape.push_back(output_attr->dims[i]); in save_npy()
|