class ngraph::op::TemporaryReplaceOutputType

Overview

Set another type for a specified output for the period of time when an instance of the class exists. More…

#include <type_relaxed.hpp>

class TemporaryReplaceOutputType
{
public:
    // construction

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

    // methods

    Output<Node> get() const;
};

Detailed Documentation

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

Construction

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

Replace element type for a given output port by tmp_type.

Methods

Output<Node> get() const

Return the output port that was used in the constructor.