Searched defs:cell_state_shape (Results 1 – 3 of 3) sorted by relevance
96 const TensorShape cell_state_shape = cell_state_in->info()->tensor_shape(); in configure() local
511 TensorShape cell_state_shape; member705 const TensorShape& cell_state_shape = model_shapes.cell_state_shape; in CreateForwardAndBackwardIODescriptors() local1591 const TensorShape& cell_state_shape = model_shapes.cell_state_shape; in AllocateOutputs() local2001 const TensorShape& cell_state_shape = model_shapes.cell_state_shape; in ExtractBackwardInputs() local2049 const TensorShape& cell_state_shape = model_shapes.cell_state_shape; in AllocateOutputs() local
116 const TensorShape cell_state_shape = cell_state_in->info()->tensor_shape(); in configure() local