昇腾社区首页
中文
注册

总体说明

  1. 统一接口的各个功能模块的API存在指针入参,如Parser模块会接收Network指针,组网模块添加及修改Layer的API,存在接收或使用Tensor、Layer指针的场景,需要注意,Network指针必须是通过Builder模块的CreateNetwork接口构造的非空指针,Layer指针必须是通过AddXXLayer的方式构造的非空指针,Tensor指针只能通过AddInput接口构造,或是通过GetOutput接口,取自合法Network或Layer的输出Tensor。使用通过其他途径构造的指针,如传递未初始化的指针会造成未定义行为,如程序的coredump。其他功能模块的指针入参,也需要遵循相同的使用规范。
  2. Layer组网期间,通过AddXXLayer构造的Layer指针,支持一定的Shape推导功能,在调用Layer->GetOutput(int index)获取输出Tensor时,接口内部会执行推导动作。须要在Layer的全部输入和属性已配置完成后再调用该接口,以便获得正确的输出Tensor和正确推导的TensorShape与DataType。另外,调用该接口后要避免再次修改Layer的属性,当前不支持再次根据新的属性重新推导输出。
  3. 当前Parser跟Network是一一对应的,即一个Parser实例只能使用一次,用户需要解析多个ONNX模型时,则也需要同时实例化多个Parser。