| Package | Description |
|---|---|
| org.apache.tvm.contrib |
| Modifier and Type | Method and Description |
|---|---|
static GraphModule |
GraphExecutor.create(String graphJson,
Module libmod,
Device dev)
Create a runtime executor module given a graph and module.
|
static GraphModule |
GraphExecutor.create(String graphJson,
Module libmod,
Device dev)
Create a runtime executor module given a graph and module.
|
GraphModule |
GraphModule.loadParams(byte[] params)
Load parameters from serialized byte array of parameter dict.
|
GraphModule |
GraphModule.loadParams(byte[] params)
Load parameters from serialized byte array of parameter dict.
|
GraphModule |
GraphModule.run()
Run forward execution of the graph.
|
GraphModule |
GraphModule.run()
Run forward execution of the graph.
|
GraphModule |
GraphModule.setInput(int key,
NDArray value)
Set inputs to the module.
|
GraphModule |
GraphModule.setInput(int key,
NDArray value)
Set inputs to the module.
|
GraphModule |
GraphModule.setInput(String key,
NDArray value)
Set inputs to the module.
|
GraphModule |
GraphModule.setInput(String key,
NDArray value)
Set inputs to the module.
|
Copyright © 2022. All rights reserved.