ai.djl.tensorflow.engine

Class TfDataType

java.lang.Object

ai.djl.tensorflow.engine.TfDataType

public final class TfDataType
extends java.lang.Object

Method Summary

Modifier and Type	Method and Description
static ai.djl.ndarray.types.DataType	fromTf(org.tensorflow.DataType<? extends org.tensorflow.types.family.TType> tfType)
static org.tensorflow.DataType<? extends org.tensorflow.types.family.TType>	toTf(ai.djl.ndarray.types.DataType type)

Methods inherited from class java.lang.Object

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

Method Detail

fromTf

public static ai.djl.ndarray.types.DataType fromTf(org.tensorflow.DataType<? extends org.tensorflow.types.family.TType> tfType)

toTf

public static org.tensorflow.DataType<? extends org.tensorflow.types.family.TType> toTf(ai.djl.ndarray.types.DataType type)