nGraph and CPU plugin opset rtti (#6445)
This commit is contained in:
parent
267f9a3b77
commit
94ee9ab7c3
@ -13,6 +13,7 @@ namespace MKLDNNPlugin {
|
||||
class FullyConnectedNode : public ngraph::op::Op {
|
||||
public:
|
||||
static constexpr ngraph::NodeTypeInfo type_info{"FullyConnected", 0};
|
||||
static constexpr const ::ngraph::Node::type_info_t& get_type_info_static() { return type_info; }
|
||||
const ngraph::NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
|
||||
FullyConnectedNode() = default;
|
||||
|
@ -11,6 +11,7 @@ namespace MKLDNNPlugin {
|
||||
class LeakyReluNode : public ngraph::op::Op {
|
||||
public:
|
||||
static constexpr ngraph::NodeTypeInfo type_info{"LeakyRelu", 0};
|
||||
static constexpr const ::ngraph::Node::type_info_t& get_type_info_static() { return type_info; }
|
||||
const ngraph::NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
|
||||
LeakyReluNode(const ngraph::Output<ngraph::Node> &data, const float &negative_slope, const ngraph::element::Type output_type);
|
||||
|
@ -11,6 +11,7 @@ namespace MKLDNNPlugin {
|
||||
class PowerStaticNode : public ngraph::op::Op {
|
||||
public:
|
||||
static constexpr ngraph::NodeTypeInfo type_info{"PowerStatic", 0};
|
||||
static constexpr const ::ngraph::Node::type_info_t& get_type_info_static() { return type_info; }
|
||||
const ngraph::NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
|
||||
PowerStaticNode(const ngraph::Output<ngraph::Node> &data, const float &power, const float &scale, const float &shift,
|
||||
|
@ -11,6 +11,7 @@ namespace MKLDNNPlugin {
|
||||
class SwishNode : public ngraph::op::Op {
|
||||
public:
|
||||
static constexpr ngraph::NodeTypeInfo type_info{"SwishCPU", 0};
|
||||
static constexpr const ::ngraph::Node::type_info_t& get_type_info_static() { return type_info; }
|
||||
const ngraph::NodeTypeInfo &get_type_info() const override { return type_info; }
|
||||
|
||||
explicit SwishNode(const ngraph::Output<Node> &input, float alpha = 1.0);
|
||||
|
@ -17,8 +17,8 @@ namespace ngraph
|
||||
class NGRAPH_API BinaryConvolution : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"BinaryConvolution", 1};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
enum class BinaryConvolutionMode
|
||||
{
|
||||
// Interpret input data and kernel values: 0 as -1, 1 as 1
|
||||
|
@ -85,8 +85,8 @@ namespace ngraph
|
||||
class NGRAPH_API Broadcast : public util::BroadcastBase
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"Broadcast", 1};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
/// \brief Constructs a broadcast operation.
|
||||
Broadcast() = default;
|
||||
/// \brief Constructs a broadcast operation.
|
||||
|
@ -26,8 +26,8 @@ namespace ngraph
|
||||
class NGRAPH_API Constant : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"Constant", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
Constant() = default;
|
||||
|
||||
/// \brief Initialize a constant from tensor
|
||||
|
@ -16,8 +16,8 @@ namespace ngraph
|
||||
class NGRAPH_API ConvertLike : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"ConvertLike", 1};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
/// \brief Constructs a conversion operation.
|
||||
ConvertLike() = default;
|
||||
/// \brief Constructs a conversion operation.
|
||||
|
@ -15,8 +15,8 @@ namespace ngraph
|
||||
class NGRAPH_API CTCGreedyDecoder : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"CTCGreedyDecoder", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
CTCGreedyDecoder() = default;
|
||||
/// \brief Constructs a CTCGreedyDecoder operation
|
||||
///
|
||||
|
@ -18,8 +18,8 @@ namespace ngraph
|
||||
class NGRAPH_API DeformableConvolution : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"DeformableConvolution", 1};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
/// \brief Constructs a conversion operation.
|
||||
DeformableConvolution() = default;
|
||||
/// \brief Constructs a conversion operation.
|
||||
|
@ -15,8 +15,8 @@ namespace ngraph
|
||||
class NGRAPH_API DeformablePSROIPooling : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"DeformablePSROIPooling", 1};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
DeformablePSROIPooling() = default;
|
||||
/// \brief Constructs a DeformablePSROIPooling operation
|
||||
///
|
||||
|
@ -37,8 +37,8 @@ namespace ngraph
|
||||
class NGRAPH_API DetectionOutput : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"DetectionOutput", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
DetectionOutput() = default;
|
||||
/// \brief Constructs a DetectionOutput operation
|
||||
///
|
||||
|
@ -20,8 +20,8 @@ namespace ngraph
|
||||
class NGRAPH_API Elu : public ngraph::op::Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"Elu", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
Elu() = default;
|
||||
/// \brief Constructs an Elu operation.
|
||||
///
|
||||
|
@ -17,8 +17,8 @@ namespace ngraph
|
||||
class NGRAPH_API GatherTree : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"GatherTree", 1};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
GatherTree() = default;
|
||||
/// \param step_ids Tensor of shape [MAX_TIME, BATCH_SIZE, BEAM_WIDTH] with
|
||||
/// indices from per each step
|
||||
|
@ -22,8 +22,8 @@ namespace ngraph
|
||||
class NGRAPH_API GRN : public ngraph::op::util::FusedOp
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"GRN", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
GRN();
|
||||
/// \brief Constructs a GRN operation.
|
||||
///
|
||||
|
@ -21,8 +21,8 @@ namespace ngraph
|
||||
class NGRAPH_API HardSigmoid : public ngraph::op::util::FusedOp
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"HardSigmoid", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
HardSigmoid();
|
||||
|
||||
/// \brief Constructs a HardSigmoid operation.
|
||||
|
@ -30,8 +30,8 @@ namespace ngraph
|
||||
class NGRAPH_API LRN : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"LRN", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
/// \brief Constructs a LRN operation.
|
||||
LRN() = default;
|
||||
/// \brief Constructs a LRN operation.
|
||||
|
@ -58,8 +58,8 @@ namespace ngraph
|
||||
class NGRAPH_API LSTMCell : public util::RNNCellBase
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"LSTMCell", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
LSTMCell();
|
||||
///
|
||||
/// \brief Constructs LSTMCell node.
|
||||
@ -280,8 +280,8 @@ namespace ngraph
|
||||
class NGRAPH_API LSTMCell : public util::RNNCellBase
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"LSTMCell", 4};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
LSTMCell();
|
||||
///
|
||||
/// \brief Constructs LSTMCell node.
|
||||
|
@ -23,8 +23,8 @@ namespace ngraph
|
||||
CENTER
|
||||
};
|
||||
|
||||
static constexpr NodeTypeInfo type_info{"NonMaxSuppression", 1};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
NonMaxSuppression() = default;
|
||||
|
||||
/// \brief Constructs a NonMaxSuppression operation.
|
||||
|
@ -15,8 +15,8 @@ namespace ngraph
|
||||
class NGRAPH_API OneHot : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"OneHot", 1};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
/// \brief Constructs a one-hot operation.
|
||||
OneHot() = default;
|
||||
/// \brief Constructs a one-hot operation.
|
||||
|
@ -18,8 +18,8 @@ namespace ngraph
|
||||
class NGRAPH_API Pad : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"Pad", 1};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
/// \brief Constructs a generic padding operation.
|
||||
///
|
||||
/// \param arg The output producing input tensor to be padded.
|
||||
|
@ -42,8 +42,8 @@ namespace ngraph
|
||||
class NGRAPH_API PriorBox : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"PriorBox", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
PriorBox() = default;
|
||||
/// \brief Constructs a PriorBox operation
|
||||
///
|
||||
|
@ -35,8 +35,8 @@ namespace ngraph
|
||||
class NGRAPH_API PriorBoxClustered : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"PriorBoxClustered", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
PriorBoxClustered() = default;
|
||||
/// \brief Constructs a PriorBoxClustered operation
|
||||
///
|
||||
|
@ -15,8 +15,8 @@ namespace ngraph
|
||||
class NGRAPH_API PSROIPooling : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"PSROIPooling", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
PSROIPooling() = default;
|
||||
/// \brief Constructs a PSROIPooling operation
|
||||
///
|
||||
|
@ -57,8 +57,8 @@ namespace ngraph
|
||||
class NGRAPH_API Range : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"Range", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
/// \brief Constructs an unitialized range operation.
|
||||
Range() = default;
|
||||
|
||||
|
@ -15,8 +15,8 @@ namespace ngraph
|
||||
class NGRAPH_API RegionYolo : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"RegionYolo", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
RegionYolo() = default;
|
||||
///
|
||||
/// \brief Constructs a RegionYolo operation
|
||||
|
@ -15,8 +15,8 @@ namespace ngraph
|
||||
class NGRAPH_API ReorgYolo : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"ReorgYolo", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
ReorgYolo() = default;
|
||||
/// \brief Constructs a ReorgYolo operation
|
||||
///
|
||||
|
@ -15,8 +15,8 @@ namespace ngraph
|
||||
class NGRAPH_API ReverseSequence : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"ReverseSequence", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
ReverseSequence() = default;
|
||||
/// \brief Constructs a ReverseSequence operation.
|
||||
///
|
||||
|
@ -41,8 +41,8 @@ namespace ngraph
|
||||
class NGRAPH_API RNNCell : public util::RNNCellBase
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"RNNCell", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
RNNCell();
|
||||
///
|
||||
/// \brief Constructs RNNCell node.
|
||||
|
@ -24,8 +24,8 @@ namespace ngraph
|
||||
class NGRAPH_API SpaceToDepth : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"SpaceToDepth", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
enum class SpaceToDepthMode
|
||||
{
|
||||
// The output depth is gathered from [block_size, ..., block_size, C]
|
||||
|
@ -20,8 +20,8 @@ namespace ngraph
|
||||
class NGRAPH_API TensorIterator : public op::util::SubGraphOp
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"TensorIterator", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
bool visit_attributes(AttributeVisitor& visitor) override;
|
||||
|
||||
TensorIterator() = default;
|
||||
|
@ -18,8 +18,8 @@ namespace ngraph
|
||||
class NGRAPH_API Tile : public Op
|
||||
{
|
||||
public:
|
||||
static constexpr NodeTypeInfo type_info{"Tile", 0};
|
||||
const NodeTypeInfo& get_type_info() const override { return type_info; }
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
Tile() = default;
|
||||
/// \brief Perform dynamic padding of a tensor
|
||||
///
|
||||
|
@ -44,6 +44,8 @@ namespace ngraph
|
||||
const BroadcastModeSpec& broadcast_mode = BroadcastType::NUMPY);
|
||||
|
||||
public:
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
void validate_and_infer_types() override;
|
||||
/// \return true and the AxisSet if broadcast axes can be fully determined.
|
||||
virtual std::pair<bool, AxisSet> get_broadcast_axes() const;
|
||||
|
@ -21,6 +21,8 @@ namespace ngraph
|
||||
: public Op
|
||||
{
|
||||
public:
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
// Fused op decomposition can be performed in the presence of
|
||||
// partial shapes
|
||||
virtual bool can_decompose_with_partial_shapes() { return false; }
|
||||
|
@ -54,6 +54,8 @@ namespace ngraph
|
||||
class NGRAPH_API RNNCellBase : public Op
|
||||
{
|
||||
public:
|
||||
NGRAPH_RTTI_DECLARATION;
|
||||
|
||||
///
|
||||
/// \brief Constructs a RNNCellBase class.
|
||||
///
|
||||
@ -87,7 +89,7 @@ namespace ngraph
|
||||
///
|
||||
void validate_input_rank_dimension(const std::vector<ngraph::PartialShape>& input);
|
||||
|
||||
virtual bool visit_attributes(AttributeVisitor& visitor);
|
||||
bool visit_attributes(AttributeVisitor& visitor) override;
|
||||
std::size_t get_hidden_size() const { return m_hidden_size; }
|
||||
float get_clip() const { return m_clip; }
|
||||
const std::vector<std::string>& get_activations() const { return m_activations; }
|
||||
|
@ -14,7 +14,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v1::BinaryConvolution::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v1::BinaryConvolution, "BinaryConvolution", 1);
|
||||
|
||||
op::v1::BinaryConvolution::BinaryConvolution(const Output<Node>& data,
|
||||
const Output<Node>& kernel,
|
||||
|
@ -245,7 +245,7 @@ namespace
|
||||
}
|
||||
} // namespace
|
||||
|
||||
constexpr NodeTypeInfo op::v1::Broadcast::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v1::Broadcast, "Broadcast", 1, op::util::BroadcastBase);
|
||||
|
||||
op::v1::Broadcast::Broadcast(const Output<Node>& arg,
|
||||
const Output<Node>& target_shape,
|
||||
|
@ -38,7 +38,7 @@ static inline string to_cpp_string(T value)
|
||||
return rc;
|
||||
}
|
||||
|
||||
constexpr NodeTypeInfo op::Constant::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::Constant, "Constant", 0);
|
||||
|
||||
op::Constant::Constant(const shared_ptr<runtime::Tensor>& tensor)
|
||||
: Constant(tensor->get_element_type(), tensor->get_shape())
|
||||
|
@ -12,7 +12,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v1::ConvertLike::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v1::ConvertLike, "ConvertLike", 1);
|
||||
|
||||
op::v1::ConvertLike::ConvertLike(const Output<Node>& data, const Output<Node>& like)
|
||||
: Op({data, like})
|
||||
|
@ -8,7 +8,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::CTCGreedyDecoder::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::CTCGreedyDecoder, "CTCGreedyDecoder", 0);
|
||||
|
||||
op::CTCGreedyDecoder::CTCGreedyDecoder(const Output<Node>& input,
|
||||
const Output<Node>& seq_len,
|
||||
|
@ -13,7 +13,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v1::DeformableConvolution::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v1::DeformableConvolution, "DeformableConvolution", 1);
|
||||
|
||||
op::v1::DeformableConvolution::DeformableConvolution(const Output<Node>& arg,
|
||||
const Output<Node>& offsets,
|
||||
|
@ -8,7 +8,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v1::DeformablePSROIPooling::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v1::DeformablePSROIPooling, "DeformablePSROIPooling", 1);
|
||||
|
||||
op::v1::DeformablePSROIPooling::DeformablePSROIPooling(const Output<Node>& input,
|
||||
const Output<Node>& coords,
|
||||
|
@ -8,7 +8,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::DetectionOutput::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::DetectionOutput, "DetectionOutput", 0);
|
||||
|
||||
op::DetectionOutput::DetectionOutput(const Output<Node>& box_logits,
|
||||
const Output<Node>& class_preds,
|
||||
|
@ -11,7 +11,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::Elu::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::Elu, "Elu", 0);
|
||||
|
||||
op::Elu::Elu(const Output<Node>& data, const double alpha)
|
||||
: Op({data})
|
||||
|
@ -9,7 +9,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v1::GatherTree::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v1::GatherTree, "GatherTree", 1);
|
||||
|
||||
op::v1::GatherTree::GatherTree(const Output<Node>& step_ids,
|
||||
const Output<Node>& parent_idx,
|
||||
|
@ -22,7 +22,7 @@ using namespace ngraph;
|
||||
|
||||
NGRAPH_SUPPRESS_DEPRECATED_START
|
||||
|
||||
constexpr NodeTypeInfo op::GRN::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::GRN, "GRN", 0, op::util::FusedOp);
|
||||
|
||||
op::GRN::GRN()
|
||||
: FusedOp()
|
||||
|
@ -18,7 +18,7 @@ using namespace ngraph;
|
||||
|
||||
NGRAPH_SUPPRESS_DEPRECATED_START
|
||||
|
||||
constexpr NodeTypeInfo op::HardSigmoid::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::HardSigmoid, "HardSigmoid", 0, op::util::FusedOp);
|
||||
|
||||
op::HardSigmoid::HardSigmoid()
|
||||
: FusedOp()
|
||||
|
@ -12,7 +12,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::LRN::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::LRN, "LRN", 0);
|
||||
|
||||
op::LRN::LRN(const Output<Node>& arg, double alpha, double beta, double bias, size_t size)
|
||||
: LRN(arg, op::Constant::create(element::i64, Shape{1}, {1}), alpha, beta, bias, size)
|
||||
|
@ -16,8 +16,8 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v4::LSTMCell::type_info;
|
||||
constexpr NodeTypeInfo op::v0::LSTMCell::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v0::LSTMCell, "LSTMCell", 0, op::util::RNNCellBase);
|
||||
NGRAPH_RTTI_DEFINITION(op::v4::LSTMCell, "LSTMCell", 4, op::util::RNNCellBase);
|
||||
|
||||
op::v0::LSTMCell::LSTMCell()
|
||||
: m_input_forget(false)
|
||||
|
@ -18,7 +18,7 @@ using namespace ngraph;
|
||||
|
||||
// ------------------------------ V1 ------------------------------
|
||||
|
||||
constexpr NodeTypeInfo op::v1::NonMaxSuppression::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v1::NonMaxSuppression, "NonMaxSuppression", 1);
|
||||
|
||||
op::v1::NonMaxSuppression::NonMaxSuppression(
|
||||
const Output<Node>& boxes,
|
||||
|
@ -12,7 +12,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v1::OneHot::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v1::OneHot, "OneHot", 1);
|
||||
|
||||
op::v1::OneHot::OneHot(const Output<Node>& indices,
|
||||
const Output<Node>& depth,
|
||||
|
@ -15,7 +15,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v1::Pad::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v1::Pad, "Pad", 1);
|
||||
|
||||
op::v1::Pad::Pad(const Output<Node>& arg,
|
||||
const Output<Node>& pads_begin,
|
||||
|
@ -14,7 +14,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::PriorBox::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::PriorBox, "PriorBox", 0);
|
||||
|
||||
op::PriorBox::PriorBox(const Output<Node>& layer_shape,
|
||||
const Output<Node>& image_shape,
|
||||
|
@ -14,7 +14,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::PriorBoxClustered::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::PriorBoxClustered, "PriorBoxClustered", 0);
|
||||
|
||||
op::PriorBoxClustered::PriorBoxClustered(const Output<Node>& layer_shape,
|
||||
const Output<Node>& image_shape,
|
||||
|
@ -9,7 +9,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::PSROIPooling::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::PSROIPooling, "PSROIPooling", 0);
|
||||
|
||||
op::PSROIPooling::PSROIPooling(const Output<Node>& input,
|
||||
const Output<Node>& coords,
|
||||
|
@ -322,7 +322,7 @@ bool op::v4::Range::has_evaluate() const
|
||||
return false;
|
||||
}
|
||||
|
||||
constexpr NodeTypeInfo op::v0::Range::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v0::Range, "Range", 0);
|
||||
|
||||
op::v0::Range::Range(const Output<Node>& start, const Output<Node>& stop, const Output<Node>& step)
|
||||
: Op({start, stop, step})
|
||||
|
@ -9,7 +9,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::RegionYolo::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::RegionYolo, "RegionYolo", 0);
|
||||
|
||||
op::RegionYolo::RegionYolo(const Output<Node>& input,
|
||||
const size_t coords,
|
||||
|
@ -9,7 +9,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::ReorgYolo::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::ReorgYolo, "ReorgYolo", 0);
|
||||
|
||||
op::ReorgYolo::ReorgYolo(const Output<Node>& input, const Strides& strides)
|
||||
: Op({input})
|
||||
|
@ -14,7 +14,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::ReverseSequence::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::ReverseSequence, "ReverseSequence", 0);
|
||||
|
||||
op::ReverseSequence::ReverseSequence(const Output<Node>& arg,
|
||||
const Output<Node>& seq_indices,
|
||||
|
@ -13,7 +13,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v0::RNNCell::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v0::RNNCell, "RNNCell", 0, util::RNNCellBase);
|
||||
|
||||
op::v0::RNNCell::RNNCell()
|
||||
{
|
||||
|
@ -18,7 +18,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::SpaceToDepth::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::SpaceToDepth, "SpaceToDepth", 0);
|
||||
|
||||
op::SpaceToDepth::SpaceToDepth(const Output<Node>& data,
|
||||
const SpaceToDepthMode& mode,
|
||||
|
@ -11,7 +11,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v0::TensorIterator::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v0::TensorIterator, "TensorIterator", 0, op::util::SubGraphOp);
|
||||
|
||||
op::v0::TensorIterator::TensorIterator(const OutputVector& values)
|
||||
: op::util::SubGraphOp(values)
|
||||
|
@ -12,7 +12,7 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
constexpr NodeTypeInfo op::v0::Tile::type_info;
|
||||
NGRAPH_RTTI_DEFINITION(op::v0::Tile, "Tile", 0);
|
||||
|
||||
op::v0::Tile::Tile(const Output<Node>& data, const Output<Node>& repeats)
|
||||
: Op({data, repeats})
|
||||
|
@ -18,6 +18,8 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
NGRAPH_RTTI_DEFINITION(op::util::BroadcastBase, "BroadcastBase", 0);
|
||||
|
||||
op::util::BroadcastBase::BroadcastBase(const Output<Node>& arg,
|
||||
const Output<Node>& target_shape,
|
||||
const Output<Node>& axes_mapping,
|
||||
|
@ -11,6 +11,8 @@ using namespace ngraph;
|
||||
|
||||
NGRAPH_SUPPRESS_DEPRECATED_START
|
||||
|
||||
NGRAPH_RTTI_DEFINITION(op::util::FusedOp, "FusedOp", 0);
|
||||
|
||||
op::util::FusedOp::FusedOp()
|
||||
: Op()
|
||||
{
|
||||
|
@ -19,6 +19,8 @@
|
||||
using namespace std;
|
||||
using namespace ngraph;
|
||||
|
||||
NGRAPH_RTTI_DEFINITION(op::util::RNNCellBase, "RNNCellBase", 0);
|
||||
|
||||
std::shared_ptr<Node> ngraph::op::util::convert_lstm_node_format(const Output<Node>& node,
|
||||
LSTMWeightsFormat from_format,
|
||||
LSTMWeightsFormat to_format,
|
||||
|
Loading…
Reference in New Issue
Block a user