Fix LSTM documentation
The documentation said that cell-to-input weights are required to be present when input-to-input weights, recurrent-to-input weights and input gate bias are present. This was incorrect since this weights can be omitted if peephole connections are not used even if all the other tensors are present. Another bug that is fixed in this change is that for output #0 the docs said "of shape [batch_size, num_units * 4] with CIFG, or [batch_size, num_units * 3] without CIFG" when in fact it is the opposite, i.e. "of shape [batch_size, num_units * 3] with CIFG, or [batch_size, num_units * 4] without CIFG." Existing CTS/VTS tests expect behaviour described in the fixed documentation. Existing CPU implementation is also compliant with the fixed documentation. Fix: 111842951 Test: mma Change-Id: Id011783e33672ae65dc6fe3784cb26feb832acf9 Merged-In: Id011783e33672ae65dc6fe3784cb26feb832acf9 (cherry picked from commit e0537f09)
Loading
Please register or sign in to comment