Produces an output by concatenating an output from each iteration. More...
#include <tensor_iterator.hpp>
Public Member Functions | |
| const type_info_t & | get_type_info () const override |
| ConcatOutputDescription (uint64_t body_value_index, uint64_t output_index, int64_t start, int64_t stride, int64_t part_size, int64_t end, int64_t axis) | |
| Constructs a new instance. More... | |
| virtual std::shared_ptr< OutputDescription > | copy () const override |
| bool | visit_attributes (AttributeVisitor &visitor) override |
Data Fields | |
| int64_t | m_start {0} |
| int64_t | m_stride {0} |
| int64_t | m_part_size {0} |
| int64_t | m_end {0} |
| int64_t | m_axis {0} |
 Data Fields inherited from ngraph::op::v0::TensorIterator::OutputDescription | |
| uint64_t | m_body_value_index {0} |
| uint64_t | m_output_index {0} |
Static Public Attributes | |
| static constexpr type_info_t | type_info {"ConcatOutputDescription", 0} |
Additional Inherited Members | |
| Public Types inherited from ngraph::op::v0::TensorIterator::OutputDescription | |
| using | type_info_t = DiscreteTypeInfo |
| Protected Member Functions inherited from ngraph::op::v0::TensorIterator::OutputDescription | |
| OutputDescription (uint64_t body_value_index, uint64_t output_index) | |
| Constructs a new instance. More... | |
Produces an output by concatenating an output from each iteration.
| ngraph::op::v0::TensorIterator::ConcatOutputDescription::ConcatOutputDescription | ( | uint64_t | body_value_index, |
| uint64_t | output_index, | ||
| int64_t | start, | ||
| int64_t | stride, | ||
| int64_t | part_size, | ||
| int64_t | end, | ||
| int64_t | axis | ||
| ) |
Constructs a new instance.
| body_value_index | A body value that produces the output |
| output_index | The TensorIterator output index |
| start | First index for slices |
| stride | Step amount for slices |
| part_size | Width of slices |
| end | Last index for slices |
| axis | Axis being sliced |
