Searched refs:hidden_state_shape (Results 1 – 1 of 1) sorted by relevance
507 TensorShape hidden_state_shape; member603 model_shapes->hidden_state_shape = in ExtractForwardInput()607 model_shapes->hidden_state_shape = in ExtractForwardInput()612 if ((*input_h)->shape() != model_shapes->hidden_state_shape) { in ExtractForwardInput()615 model_shapes->hidden_state_shape.DebugString()); in ExtractForwardInput()660 const TensorShape& hidden_state_shape = model_shapes.hidden_state_shape; in CreateForwardAndBackwardIODescriptors() local686 DCHECK_EQ(hidden_state_shape.dims(), 3); in CreateForwardAndBackwardIODescriptors()689 hidden_state_shape.dim_size(0), hidden_state_shape.dim_size(1), in CreateForwardAndBackwardIODescriptors()690 hidden_state_shape.dim_size(2), data_type); in CreateForwardAndBackwardIODescriptors()695 hidden_state_shape.dim_size(1), hidden_state_shape.dim_size(0), in CreateForwardAndBackwardIODescriptors()[all …]