Class: GraphModule

tvm.GraphModule()

new GraphModule()

Wrapper runtime module. Wraps around set_input, load_params, run, and get_output.
Source:

Methods

get_output(out) → {NDArray}

Get index-th output to out.
Parameters:
Name Type Description
out NDArray The output array container.
Source:
Returns:
The output array container.
Type
NDArray

load_base64_params(base64_params)

Load parameters from serialized base64 string of parameter dict.
Parameters:
Name Type Description
base64_params string The serialized parameter dict.
Source:

load_params(params)

Load parameters from serialized byte array of parameter dict.
Parameters:
Name Type Description
params Uint8Array The serialized parameter dict.
Source:

run()

Run forward execution of the graph.
Source:

set_input(key, value)

Set input to graph module.
Parameters:
Name Type Description
key string The name of the input.
value NDArray The input value.
Source: