ELMTWISE_INFER_SHAPEANDTYPE

Description

Encapsulated macro of common functions, facilitating the development of the InferShape function. This API sets the output shape and dtype based on the input shape and dtype.

For example, if the input shape is (1, 2, 3, 4) and input type is float, the macro sets the output shape to (1, 2, 3, 4) and the output dtype to float.

Prototype

ELMTWISE_INFER_SHAPEANDTYPE(in_name, out_name)

Restrictions

None

Parameters

Returns

Success or failure

Example

COMMON_INFER_FUNC_REG(DiagD, ELMTWISE_INFER_SHAPEANDTYPE("assist", "y"));