Describes a body input formed from slices of an input to TensorIterator. More...
#include <tensor_iterator.hpp>
Public Member Functions | |
| const type_info_t & | get_type_info () const override |
| SliceInputDescription (uint64_t input_index, uint64_t body_parameter_index, int64_t start, int64_t stride, int64_t part_size, int64_t end, int64_t axis) | |
| Constructs a new instance. More... | |
| std::shared_ptr< InputDescription > | 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::InputDescription | |
| uint64_t | m_input_index {0} |
| uint64_t | m_body_parameter_index {0} |
Static Public Attributes | |
| static constexpr type_info_t | type_info {"SliceInputDescription", 0} |
Additional Inherited Members | |
| Public Types inherited from ngraph::op::v0::TensorIterator::InputDescription | |
| using | type_info_t = DiscreteTypeInfo |
| Protected Member Functions inherited from ngraph::op::v0::TensorIterator::InputDescription | |
| InputDescription (uint64_t input_index, uint64_t body_parameter_index) | |
| Constructs a new instance. More... | |
Describes a body input formed from slices of an input to TensorIterator.
| ngraph::op::v0::TensorIterator::SliceInputDescription::SliceInputDescription | ( | uint64_t | input_index, |
| uint64_t | body_parameter_index, | ||
| int64_t | start, | ||
| int64_t | stride, | ||
| int64_t | part_size, | ||
| int64_t | end, | ||
| int64_t | axis | ||
| ) |
Constructs a new instance.
| input_index | Position of the TensorIterator input |
| body_parameter_index | Body parameter position to receive input |
| start | First index for slices |
| stride | Step amount for slices |
| part_size | Width of slices |
| end | Last index for slices |
| axis | Axis being sliced |
