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

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