TensorTypeRegistry

public final class TensorTypeRegistry

Repository of all registered tensor types.

Public Constructors

Public Methods

static <T extends TType> TensorTypeInfo<T>
find(DataType dataType)
Find registered information about a tensor type from its equivalent data type
static <T extends TType> TensorTypeInfo<T>
find(Class<T> type)
Find registered information about a tensor type from its class

Inherited Methods

boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

Public Constructors

public TensorTypeRegistry ()

Public Methods

public static TensorTypeInfo<T> find (DataType dataType)

Find registered information about a tensor type from its equivalent data type

Parameters
dataType data type
Returns
  • type registered information
Throws
IllegalArgumentException if no tensor type for this data type has been registered

public static TensorTypeInfo<T> find (Class<T> type)

Find registered information about a tensor type from its class

Parameters
type class implementing TType
Returns
  • type registered information
Throws
IllegalArgumentException if the provided class has not been registered as a tensor type