更新时间:2021-03-18 GMT+08:00
分享

AIModelManager::IsPreAllocateOutputMem

是否可以预分配输出内存。

相关函数:

AIStatus AIModelManager::CreateOutputTensor(const std::vector<std::shared_ptr<IAITensor>> &in_data, std::vector<std::shared_ptr<IAITensor>> &out_data);

函数格式

virtual bool AIModelManager::IsPreAllocateOutputMem() override ;

参数说明

无。

返回值

  • True:可调用CreateOutputTensor申请输出内存申请。
  • False:不可调用CreateOutputTensor。

    只有模型管家加载一个模型时才返回true。

分享:

    相关文档

    相关产品