Searched refs:state_shape (Results 1 – 6 of 6) sorted by relevance
600 state_shape = state.get_shape()601 assert state_shape is not None602 assert state_shape.ndims == 2603 assert state_shape[0] == state_shape[1]604 dimension = state_shape.dims[0].value
418 state_shape = self.compute_output_shape(input_shape)420 state_shape = state_shape[0]422 state_shape = state_shape[:1].concatenate(state_shape[2:])423 if None in state_shape:439 result = list(state_shape)
419 state_shape = output_shape[1:]431 return output_shape + state_shape + copy.copy(state_shape)432 return [output_shape] + state_shape + copy.copy(state_shape)
471 state_shape = [batch] + tensor_shape.as_shape(flat_state).as_list()472 return tensor_shape.as_shape(state_shape)473 state_shape = nest.map_structure(_get_state_shape, state_size)474 return generic_utils.to_list(output_shape) + nest.flatten(state_shape)
440 def state_shape(self, batch_size): member in _CudnnRNN445 for sp in self.state_shape(batch_size):547 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() local2429 const int kDims = state_shape.dimensions_count(); in AddPlaceholderForRNNState()2431 shape->add_dim()->set_size(state_shape.dims(i)); in AddPlaceholderForRNNState()