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