InternalMixedPrecisionModule
constructor(ioType: TIO, internalType: TInternal, conversionOps: MixedPrecisionTensorOps<V>)(source)
Parameters
TIO
The input/output precision type
TInternal
The internal computation precision type
V
The value type corresponding to the DType