Searched defs:zero_state (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ | ||
H A D | rnn_cell_wrapper_impl.py | 197 def zero_state(self, batch_size, dtype): member in DropoutWrapperBase |
H A D | rnn_cell_impl.py | 313 def zero_state(self, batch_size, dtype): member in RNNCell |
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ | ||
H A D | rnn_test.py | 73 def zero_state(self, batch_size, dtype): member in ScalarStateRNNCell |
H A D | rnn_cell_test.py | 3225 def zero_state(self, batch_size=None, dtype=None): member in DropoutWrapperTest.testDropoutWrapperZeroState._Cell |