aclmdlQuerySize
函数功能
根据模型文件获取模型执行时所需的权值内存大小、工作内存大小。
约束说明
工作和权重内存为Device内存,而且需要用户申请和释放。
当由用户管理内存时,为确保内存不浪费,在申请工作内存、权值内存前,需要调用本接口查询模型运行时所需工作内存、权值内存的大小。
如果模型输入数据的Shape不确定,则不能调用aclmdlQuerySize接口查询内存大小,在加载模型时,就无法由用户管理内存,因此需选择由系统管理内存的模型加载接口(例如,aclmdlLoadFromFile、aclmdlLoadFromMem)。
函数原型
aclError aclmdlQuerySize(const char *fileName, size_t *workSize, size_t *weightSize)
参数说明
返回值说明
返回0表示成功,返回其他值表示失败。
父主题: 模型执行