megengine.functional.external.extern_opr_subgraph

extern_opr_subgraph(inputs, output_shapes, dump_name, dump_data, output_dtypes)[源代码]

加载序列化的extern opr子图并假执行运算符。

参数

  • inputs – 输入tensors列表

  • output_shapes (List[tuple]) – 输出形状

  • dump_name (str) – 子图名字

  • dump_data (bytes) – 序列化子图。

上一页

megengine.functional.distributed.scatter

下一页

megengine.functional.external.tensorrt_runtime_opr