ai.djl.mxnet.jna

Class FunctionInfo

java.lang.Object

ai.djl.mxnet.jna.FunctionInfo

public class FunctionInfo
extends java.lang.Object

A FunctionInfo represents an operator (ie function) within the MXNet Engine.

Method Summary

Modifier and Type	Method and Description
java.util.List<java.lang.String>	getArgumentNames() Returns the names of the params to the operator.
java.util.List<java.lang.String>	getArgumentTypes() Returns the types of the operator arguments.
java.lang.String	getFunctionName() Returns the name of the operator.
int	invoke(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray[] src, ai.djl.ndarray.NDArray[] dest, ai.djl.util.PairList<java.lang.String,?> params) Calls an operator with the given arguments.
ai.djl.ndarray.NDArray[]	invoke(ai.djl.ndarray.NDManager manager, ai.djl.ndarray.NDArray[] src, ai.djl.util.PairList<java.lang.String,?> params) Calls an operator with the given arguments.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

invoke

public int invoke(ai.djl.ndarray.NDManager manager,
                  ai.djl.ndarray.NDArray[] src,
                  ai.djl.ndarray.NDArray[] dest,
                  ai.djl.util.PairList<java.lang.String,?> params)

Calls an operator with the given arguments.

Parameters:

manager - the manager to attach the result to

src - the input NDArray(s) to the operator

dest - the destination NDArray(s) to be overwritten with the result of the operator

params - the non-NDArray arguments to the operator. Should be a PairList<String, String>

Returns:

the error code or zero for no errors

invoke

public ai.djl.ndarray.NDArray[] invoke(ai.djl.ndarray.NDManager manager,
                                       ai.djl.ndarray.NDArray[] src,
                                       ai.djl.util.PairList<java.lang.String,?> params)

Calls an operator with the given arguments.

Parameters:

manager - the manager to attach the result to

src - the input NDArray(s) to the operator

params - the non-NDArray arguments to the operator. Should be a PairList<String, String>

Returns:

the error code or zero for no errors

getFunctionName

public java.lang.String getFunctionName()

Returns the name of the operator.

Returns:

the name of the operator

getArgumentNames

public java.util.List<java.lang.String> getArgumentNames()

Returns the names of the params to the operator.

Returns:

the names of the params to the operator

getArgumentTypes

public java.util.List<java.lang.String> getArgumentTypes()

Returns the types of the operator arguments.

Returns:

the types of the operator arguments