Searched refs:save_npy (Results 1 – 4 of 4) sorted by relevance
| /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() function
464 save_npy(output_path, (float*)outputs[i].buf, &output_attrs[i]); in main()
|
| /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() function
581 save_npy<float>(output_path, (float*)outputs[i].buf, &cur_output_attrs[i]); in main()
|
| 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() function
627 save_npy<float>(output_path, (float*)output_mems[i]->virt_addr, &cur_output_attrs[i]); in main()
|
| /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() function
|