ScopeBasicLSTMCellPass
功能说明
将tf.nn.rnn_cell.BasicLSTMCell生成的Scope内的小算子组合融合为BasicLSTMCell算子。
Scope详情
融合对应关系
Concat算子的第1个输入作为融合后的第1个输入x。
Concat算子的第2个输入作为融合后的第2个输入h。
Mul算子的第1个输入作为融合后的第3个输入c。
MatMul算子的第2个输入作为融合后的第4个输入w。
BiasAdd算子的第2个输入作为融合后的第5个输入b。
Add_1的输出作为融合后的第0个输出ct。
Mul_2的输出作为融合后的第1个输出ht。
适用网络
使用BasicLSTMCell单cell的非循环推理网络，例如NMT推理网络。
融合规则类型
定制化融合规则
父主题： 融合规则说明