megengine.functional.external.cambricon_runtime_opr

cambricon_runtime_opr(inputs, data, symbol, tensor_dim_mutable)[源代码]

将一个序列化后的寒武纪模型加载成 MegEngine 中的一个运行时算子。

参数

  • inputs – 由输入 tensor 组成的列表。

  • data – 序列化的寒武纪模型。

  • symbol – 寒武纪模型中的函数名称。

  • tensor_dim_mutable – cnrtModel_t 中输入 tensor 的形状是否可变。

上一页

megengine.functional.external.tensorrt_runtime_opr

下一页

megengine.functional.external.atlas_runtime_opr