Data Fields
ngraph::pass::ConvertTensorIteratorToGRUSequence Class Reference

Finds all TensorIterator layers, detects the pattern Squeeze->GRUCell->Unsqueeze in the TensorIterator body, converts this pattern to GRUSequence layer and replaces them TensorIterator. More...

#include <convert_ti_to_sequences.hpp>

Inheritance diagram for ngraph::pass::ConvertTensorIteratorToGRUSequence:

Data Fields

 NGRAPH_RTTI_DECLARATION
 

Detailed Description

Finds all TensorIterator layers, detects the pattern Squeeze->GRUCell->Unsqueeze in the TensorIterator body, converts this pattern to GRUSequence layer and replaces them TensorIterator.


The documentation for this class was generated from the following file: