InferOutDataTypeSameWithFirstInput

Description

Registers a datatype deduction rule that uses the first input datatype of the operator as datatype of all outputs.

Prototype

OpImplRegisterV2 &InferOutDataTypeSameWithFirstInput()

Parameters

None

Returns

OpImplRegisterV2 object of the operator, where the datatype deduction rule is registered.

Restrictions

• You can choose to register this rule or the custom deduction rule. If both InferDataType and InferOutDataTypeByFirstInput are registered, the last registered rule is enabled.
• If the operator has no input or datatype of the first input is undefined (DT_UNDEFINED), an error is reported during deduction.