Class ov::op::TemporaryReplaceOutputType

class TemporaryReplaceOutputType

Set another type for a specified output for the period of time when an instance of the class exists. When the execution leaves the scope where an onject of TemporaryReplaceOutputType is defined, the type of the output is set to its original value. Used when initialized TypeRelaxed<BaseOp> operation in case when inputs have types that are not compatible with BaseOp infer function. In this case before TypeRelaxed is constructed the BaseOp contructor requires modified data types. So it should be

Public Functions

TemporaryReplaceOutputType(Output<Node> output, element::Type tmp_type)

Replace element type for a given output port by tmp_type.

Output<Node> get() const

Return the output port that was used in the constructor.

~TemporaryReplaceOutputType()

Restores the original element type for the output.