ai.djl.mxnet.engine

Class MxOpParams

java.lang.Object

ai.djl.util.PairList<java.lang.String,java.lang.Object>

ai.djl.mxnet.engine.MxOpParams

All Implemented Interfaces:

java.lang.Iterable<ai.djl.util.Pair<java.lang.String,java.lang.Object>>

public class MxOpParams
extends ai.djl.util.PairList<java.lang.String,java.lang.Object>

An internal helper for creating the MXNet operator parameters.

Constructor Summary

Constructors

Constructor and Description
MxOpParams()

Method Summary

Modifier and Type	Method and Description
void	addParam(java.lang.String paramName, boolean value) Adds a parameter.
void	addParam(java.lang.String paramName, double value) Adds a parameter.
void	addParam(java.lang.String paramName, float value) Adds a parameter.
void	addParam(java.lang.String paramName, int value) Adds a parameter.
void	addParam(java.lang.String paramName, long value) Adds a parameter.
void	addParam(java.lang.String paramName, java.lang.Number value) Adds a parameter.
void	addParam(java.lang.String paramName, ai.djl.ndarray.types.Shape shape) Adds a parameter.
void	addParam(java.lang.String paramName, java.lang.String value) Adds a parameter.
void	addTupleParam(java.lang.String paramName, float... tuple) Adds a parameter with tuple value.
void	addTupleParam(java.lang.String paramName, int... tuple) Adds a parameter with tuple value.
void	addTupleParam(java.lang.String paramName, long... tuple) Adds a parameter with tuple value.
void	setDataType(ai.djl.ndarray.types.DataType dataType) Sets the dataType to use for the operation.
void	setDevice(ai.djl.Device device) Sets the device to use for the operation.
void	setParam(java.lang.String paramName, java.lang.String value) Sets a (potentially existing) parameter to a new value.
void	setShape(ai.djl.ndarray.types.Shape shape) Sets the Shape parameter.
void	setSparseFormat(ai.djl.ndarray.types.SparseFormat sparseFormat) Sets the sparseFormat to use for the operation.

Methods inherited from class ai.djl.util.PairList

add, add, add, addAll, contains, get, isEmpty, iterator, keyArray, keyAt, keys, remove, remove, size, stream, subList, subList, toMap, toMap, unique, valueArray, valueAt, values

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.lang.Iterable

forEach, spliterator

Constructor Detail

MxOpParams

public MxOpParams()

Method Detail

setShape

public void setShape(ai.djl.ndarray.types.Shape shape)

Sets the Shape parameter.

Parameters:

shape - the shape to set

setDevice

public void setDevice(ai.djl.Device device)

Sets the device to use for the operation.

Parameters:

device - the device to use for the operation

setDataType

public void setDataType(ai.djl.ndarray.types.DataType dataType)

Sets the dataType to use for the operation.

Parameters:

dataType - the dataType to use for the operation

setSparseFormat

public void setSparseFormat(ai.djl.ndarray.types.SparseFormat sparseFormat)

Sets the sparseFormat to use for the operation.

Parameters:

sparseFormat - the sparseFormat to use for the operation

setParam

public void setParam(java.lang.String paramName,
                     java.lang.String value)

Sets a (potentially existing) parameter to a new value.

Parameters:

paramName - the parameter name to update

value - the value to set the parameter to

addParam

public void addParam(java.lang.String paramName,
                     ai.djl.ndarray.types.Shape shape)

Adds a parameter.

Parameters:

paramName - the name of the new parameter

shape - the value of the new parameter

addParam

public void addParam(java.lang.String paramName,
                     java.lang.String value)

Adds a parameter.

Parameters:

paramName - the name of the new parameter

value - the value of the new parameter

addParam

public void addParam(java.lang.String paramName,
                     int value)

Adds a parameter.

Parameters:

paramName - the name of the new parameter

value - the value of the new parameter

addParam

public void addParam(java.lang.String paramName,
                     long value)

Adds a parameter.

Parameters:

paramName - the name of the new parameter

value - the value of the new parameter

addParam

public void addParam(java.lang.String paramName,
                     double value)

Adds a parameter.

Parameters:

paramName - the name of the new parameter

value - the value of the new parameter

addParam

public void addParam(java.lang.String paramName,
                     float value)

Adds a parameter.

Parameters:

paramName - the name of the new parameter

value - the value of the new parameter

addParam

public void addParam(java.lang.String paramName,
                     boolean value)

Adds a parameter.

Parameters:

paramName - the name of the new parameter

value - the value of the new parameter

addParam

public void addParam(java.lang.String paramName,
                     java.lang.Number value)

Adds a parameter.

Parameters:

paramName - the name of the new parameter

value - the value of the new parameter

addTupleParam

public void addTupleParam(java.lang.String paramName,
                          int... tuple)

Adds a parameter with tuple value.

Parameters:

paramName - the name of the new parameter

tuple - the values of the new parameter

addTupleParam

public void addTupleParam(java.lang.String paramName,
                          long... tuple)

Adds a parameter with tuple value.

Parameters:

paramName - the name of the new parameter

tuple - the values of the new parameter

addTupleParam

public void addTupleParam(java.lang.String paramName,
                          float... tuple)

Adds a parameter with tuple value.

Parameters:

paramName - the name of the new parameter

tuple - the values of the new parameter