75 lines
2.4 KiB
C++
75 lines
2.4 KiB
C++
// Copyright (C) 2018-2025 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include "openvino/op/util/assign_base.hpp"
|
|
#include "openvino/op/util/variable.hpp"
|
|
|
|
namespace ov {
|
|
namespace op {
|
|
namespace v3 {
|
|
/// \brief Assign operation sets an input value to the variable with `variable_id`
|
|
/// \ingroup ov_ops_cpp_api
|
|
class OPENVINO_API Assign : public util::AssignBase {
|
|
public:
|
|
OPENVINO_OP("Assign", "opset3", util::AssignBase);
|
|
Assign() = default;
|
|
|
|
/// \brief Constructs an Assign operation.
|
|
///
|
|
/// \param new_value Node that produces the input tensor.
|
|
/// \param variable_id identifier of the variable to be updated.
|
|
Assign(const Output<Node>& new_value, const std::string& variable_id);
|
|
|
|
void validate_and_infer_types() override;
|
|
std::string get_variable_id() const override {
|
|
return m_variable_id;
|
|
}
|
|
|
|
std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const override;
|
|
|
|
bool visit_attributes(AttributeVisitor& visitor) override;
|
|
|
|
private:
|
|
std::string m_variable_id;
|
|
};
|
|
} // namespace v3
|
|
|
|
namespace v6 {
|
|
/// \brief Assign operation sets an input value to the variable with `variable_id`
|
|
/// \ingroup ov_ops_cpp_api
|
|
class OPENVINO_API Assign : public util::AssignBase {
|
|
public:
|
|
OPENVINO_OP("Assign", "opset6", util::AssignBase);
|
|
Assign() = default;
|
|
|
|
/// \brief Constructs an Assign operation.
|
|
///
|
|
/// \param new_value Node that produces the input tensor.
|
|
/// \param variable Class for storing and synchronizing element types, shapes and
|
|
/// identifiers
|
|
/// between pairs of Assign/ReadValue nodes.
|
|
Assign(const Output<Node>& new_value, const std::shared_ptr<util::Variable>& variable);
|
|
|
|
void validate_and_infer_types() override;
|
|
|
|
std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const override;
|
|
|
|
bool visit_attributes(AttributeVisitor& visitor) override;
|
|
|
|
std::string get_variable_id() const override {
|
|
OPENVINO_ASSERT(m_variable, "Variable is not initialized. Variable_id is unavailable");
|
|
return m_variable->get_info().variable_id;
|
|
}
|
|
bool evaluate(TensorVector& outputs,
|
|
const TensorVector& inputs,
|
|
const EvaluationContext& evaluation_context) const override;
|
|
bool has_evaluate() const override;
|
|
bool can_constant_fold(const OutputVector& inputs_values) const override;
|
|
};
|
|
} // namespace v6
|
|
} // namespace op
|
|
} // namespace ov
|