rml_op_cast_params

Parameters for the RML_OP_CAST operation.

Performs the cast function f(x) = cast(x), casting each element in the input to the data type of the output tensor.

API

Graph Manipulation

Items

Name

Type

Description

input

rml_op

Operation that outputs tensor containing input data.

cast_to

rml_dtype

Output data type.