Framework Bridge APIs¶
MXNet bridge wrap Function MXNet’s async function.
Wrap a TVM function as MXNet function
MXNet function runs asynchrously via its engine.
func (Function) – A TVM function that can take positional arguments
const_loc (list of int) – List of integers indicating the argument position of read only NDArray argument. The NDArray argument location that are not annotated will be viewed as mutable arrays in MXNet’s engine.
async_func – A function that can take MXNet NDArray as argument in places that used to expect TVM NDArray. Run asynchrously in MXNet’s async engine.
- Return type