更新时间:2021-03-18 GMT+08:00
AIAPIDescription
API描述,主要用于描述API的名称、输入tensor、输出tensor等,详细实现请参考ai_types.proto。
message AIAPIDescription { string name = 1; // API名称 string desc = 2; // API描述 bool isPreAllocateOutputMem = 3; // 是否预分配输出内存 AIConfig config = 4; // 配置参数 repeated AITensorDescription inputs = 5; // 输入Tensor描述 repeated AITensorDescription outputs = 6; // 输出Tensor描述 bool need_verify = 7; // 串联时是否需要校验Tensor匹配 repeated string ignored_check_aitensor = 8; // 指定在串联时不与Inputs的Tensor做匹配校验的Tensor列表 };
父主题: 数据类型