ScopeBasicLSTMCellPass
Description
Fuses the small operators within the scope generated by tf.nn.rnn_cell.BasicLSTMCell into a BasicLSTMCell operator.
Scope Details
Result Operator Prototype
BasicLSTMCell. For details, see Operator Acceleration Library API Reference.
Fusion Mapping
Input 1 of the concat operator is used as input 1 x after fusion.
Input 2 of the concat operator is used as input 2 h after fusion.
Input 1 of the Mul operator is used as input 3 c after fusion.
Input 2 of the MatMul operator is used as input 4 w after fusion.
Input 2 of the BiasAdd operator is used as input 5 b after fusion.
The output of Add_1 is used as output 0 ct after fusion.
The output of Mul_2 is used as the output 1 ht after fusion.
Applicable Network
Non-loop inference network that uses a single BasicLSTMCell, such as the NMT
Fusion Pattern Type
Non-general fusion pattern
Parent topic: Fusion Patterns