45 lines
1.2 KiB
C++
45 lines
1.2 KiB
C++
// Copyright (C) 2018-2025 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <vector>
|
|
|
|
#include "openvino/core/model.hpp"
|
|
#include "openvino/op/parameter.hpp"
|
|
#include "openvino/op/util/sub_graph_base.hpp"
|
|
|
|
namespace ov {
|
|
namespace op {
|
|
namespace v0 {
|
|
/// \brief Iterate a body over tensors, accumulating into tensors.
|
|
/// \ingroup ov_ops_cpp_api
|
|
class OPENVINO_API TensorIterator : public op::util::SubGraphOp {
|
|
public:
|
|
OPENVINO_OP("TensorIterator", "opset1", op::util::SubGraphOp);
|
|
|
|
bool visit_attributes(AttributeVisitor& visitor) override;
|
|
|
|
TensorIterator() = default;
|
|
explicit TensorIterator(const OutputVector& values);
|
|
|
|
std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const override;
|
|
/// \return the body of the iteration
|
|
std::shared_ptr<Model> get_body() const {
|
|
return m_bodies[0];
|
|
}
|
|
/// \param body set the body of the iteration
|
|
void set_body(const std::shared_ptr<Model>& body) {
|
|
set_function(body);
|
|
}
|
|
void validate_and_infer_types() override;
|
|
void revalidate_and_infer_types_for_body_ops();
|
|
|
|
private:
|
|
void try_to_set_num_iterations_if_no_slice_inputs();
|
|
};
|
|
} // namespace v0
|
|
} // namespace op
|
|
} // namespace ov
|