ngraph.utils.tensor_iterator_types

Helper classes for aggregating TensorIterator input/output desciptor attributes.

Module Attributes

Functions

Classes

GraphBody(parameters, results)

Class containing graph parameters and results.

Node

ngraph.impl.Node wraps ngraph::Node

Parameter

ngraph.impl.op.Parameter wraps ngraph::op::Parameter

TensorIteratorBodyOutputDesc(body_value_idx, …)

Represents an output from a specific iteration.

TensorIteratorConcatOutputDesc(…)

Represents an output produced by concatenation of output from each iteration.

TensorIteratorInputDesc(input_idx, …)

Represents a generic input descriptor for TensorIterator operator.

TensorIteratorInvariantInputDesc(input_idx, …)

Represents a TensorIterator graph body input that has invariant value during iteration.

TensorIteratorMergedInputDesc(input_idx, …)

Represents a TensorIterator graph body input with initial value in the first iteration.

TensorIteratorOutputDesc(body_value_idx, …)

Represents a generic output descriptor for TensorIterator operator.

TensorIteratorSliceInputDesc(input_idx, …)

Represents a TensorIterator graph body input formed from slices of TensorIterator input.

Exceptions

Modules