88 lines
2.9 KiB
C++
88 lines
2.9 KiB
C++
|
// Copyright (C) 2018-2025 Intel Corporation
|
||
|
|
// SPDX-License-Identifier: Apache-2.0
|
||
|
|
//
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "openvino/op/util/read_value_base.hpp"
|
||
|
|
#include "openvino/op/util/variable.hpp"
|
||
|
|
|
||
|
|
namespace ov {
|
||
|
|
namespace op {
|
||
|
|
namespace v3 {
|
||
|
|
/// \brief ReadValue operation creates the variable with `variable_id` and returns value
|
||
|
|
/// of this variable.
|
||
|
|
/// \ingroup ov_ops_cpp_api
|
||
|
|
class OPENVINO_API ReadValue : public util::ReadValueBase {
|
||
|
|
public:
|
||
|
|
OPENVINO_OP("ReadValue", "opset3", util::ReadValueBase);
|
||
|
|
ReadValue() = default;
|
||
|
|
|
||
|
|
/// \brief Constructs a ReadValue operation.
|
||
|
|
///
|
||
|
|
/// \param init_value Node that produces the input tensor.
|
||
|
|
/// \param variable_id identificator of the variable to create.
|
||
|
|
ReadValue(const Output<Node>& init_value, const std::string& variable_id);
|
||
|
|
|
||
|
|
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 {
|
||
|
|
return m_variable_id;
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
std::string m_variable_id;
|
||
|
|
};
|
||
|
|
} // namespace v3
|
||
|
|
|
||
|
|
namespace v6 {
|
||
|
|
/// \brief ReadValue operation gets an input value from the variable with `variable_id`
|
||
|
|
/// and returns it as an output.
|
||
|
|
/// \ingroup ov_ops_cpp_api
|
||
|
|
class OPENVINO_API ReadValue : public util::ReadValueBase {
|
||
|
|
public:
|
||
|
|
OPENVINO_OP("ReadValue", "opset6", util::ReadValueBase);
|
||
|
|
ReadValue() = default;
|
||
|
|
|
||
|
|
/// \brief Constructs a ReadValue operation.
|
||
|
|
///
|
||
|
|
/// \param variable Class for storing and synchronizing element types, shapes and
|
||
|
|
/// identifiers between pairs of Assign/ReadValue nodes.
|
||
|
|
explicit ReadValue(const std::shared_ptr<util::Variable>& variable);
|
||
|
|
|
||
|
|
/// \brief Constructs a ReadValue operation.
|
||
|
|
///
|
||
|
|
/// \param init_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.
|
||
|
|
ReadValue(const Output<Node>& init_value, const std::shared_ptr<util::Variable>& variable);
|
||
|
|
|
||
|
|
void validate_and_infer_types() override;
|
||
|
|
|
||
|
|
void revalidate_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
|