Searched refs:weight_mems (Results 1 – 1 of 1) sorted by relevance
343 rknn_tensor_mem *weight_mems[TOTAL_RKNN_MODEL_NUM]; in main() local472 weight_mems[n] = rknn_create_mem(ctx[n], mem_size[n].total_weight_size); in main()473 rknn_set_weight_mem(ctx[n], weight_mems[n]); in main()620 if (weight_mems[n]) in main()622 rknn_destroy_mem(ctx[n], weight_mems[n]); in main()