ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

CastOp

public class CastOp

Casts a TensorBuffer to a specified data type.

Public Constructors

CastOp(DataType destinationType)
Constructs a CastOp.

Public Methods

TensorBuffer

Inherited Methods

Public Constructors

public CastOp (DataType destinationType)

Constructs a CastOp.

Note: For only converting type for a certain TensorBuffer on-the-fly rather than in a processor, please directly use TensorBuffer.createFrom(TensorBuffer, DataType).

When this Op is executed, if the original TensorBuffer is already in destinationType, the original buffer will be directly returned.

Parameters
destinationType The type of the casted TensorBuffer.
Throws
IllegalArgumentException if destinationType is neither DataType.UINT8 nor DataType.FLOAT32.

Public Methods

public TensorBuffer apply (TensorBuffer input)

Parameters
input