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

AIModelDescription

调用Init接口时作为输入,用于描述模型,详细实现参考ai_types.proto。

messAIPPDynaage AIModelDescription
{
    string name = 1;  // 模型名称,支持大小写字母、数字、下划线和点(.)
    int32 type = 2;   // 模型类型,当前仅支持 DAVINCI_OFFLINE_MODEL类型,值为0
                      // 模型管家已新增模型解析能力,因此该字段不设置亦无碍,为了保持向前兼容,特此保留--2018/11/24
    string version = 3;  // 模型版本
    int32 size = 4;  // 模型大小
    string path = 5;  // 模型路径,支持大小写字母、数字、下划线
    repeated string sub_path = 6;  // 辅助模型路径,用于模型文件为多个的情况,比如caffe在线模型
                                  // 为了向前兼容,我们没有对path字段做修改(修改为repeated),而是新增了sub_path字段
    string key = 7;  // 模型秘钥
    repeated string sub_key = 8; // 辅助模型秘钥,用于模型秘钥为多个的情况,比如caffe在线模型
                                 // 为了向前兼容,我们没有对key字段做修改(修改为repeated),而是新增了sub_key字段
    enum Frequency   
    {
        UNSET   =0;   
        LOW     =1;   
        MEDIUM  =2;  
        HIGH    =3;   
    }
    Frequency frequency = 9;
    enum DeviceType  
    {
        NPU  = 0;   
        IPU  = 1;   
        MLU  = 2;   
        CPU  = 3;
        NONE = 255;
    }
    DeviceType device_type = 10;
    enum Framework  
    {
        OFFLINE    =0;
        CAFFE      =1;
        TENSORFLOW =2;
    }
    Framework framework = 11;
    bytes data = 100;  // 模型数据
    repeated AITensorDescription inputs = 12;  // 输入Tensor描述
    repeated AITensorDescription outputs = 13; // 输出Tensor描述
};
分享:

    相关文档

    相关产品