megengine.functional.external.extern_opr_subgraph

extern_opr_subgraph(inputs, output_shapes, dump_name, dump_data, output_dtypes)[source]

Load a serialized extern opr subgraph and fake execute the operator.

Parameters

  • inputs – list of input tensors.

  • output_shapes (List[tuple]) – The output shapes.

  • dump_name (str) – The serialized subgraph name.

  • dump_data (bytes) – The serialized subgraph.

previous

megengine.functional.distributed.scatter

next

megengine.functional.external.tensorrt_runtime_opr