Searched defs:state_shape (Results 1 – 2 of 2) sorted by relevance
440 def state_shape(self, batch_size): member in _CudnnRNN547 def state_shape(self, batch_size): member in CudnnLSTM574 def state_shape(self, batch_size): member in _CudnnRNNNoInputC
2428 const auto& state_shape = state_array.shape(); in AddPlaceholderForRNNState() local