21 #include "ngraph/function.hpp"
22 #include "ngraph/op/parameter.hpp"
23 #include "ngraph/op/util/sub_graph_base.hpp"
35 static constexpr
NodeTypeInfo type_info{
"TensorIterator", 0};
43 clone_with_new_inputs(
const OutputVector& new_args)
const override;
45 std::shared_ptr<Function>
get_body()
const {
return m_body; }
47 void set_body(
const std::shared_ptr<Function>& body) { m_body = body; }
49 void revalidate_and_infer_types_for_body_ops();
54 void try_to_set_num_iterations_if_no_slice_inputs();
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
Abstract base class for sub-graph based ops, i.e ops that have sub-graph.
Definition: sub_graph_base.hpp:31
Iterate a body over tensors, accumulating into tensors.
Definition: tensor_iterator.hpp:33
void set_body(const std::shared_ptr< Function > &body)
Definition: tensor_iterator.hpp:47
const NodeTypeInfo & get_type_info() const override
Definition: tensor_iterator.hpp:36
std::shared_ptr< Function > get_body() const
Definition: tensor_iterator.hpp:45
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
std::shared_ptr< Function > get_function() override
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28