Home
last modified time | relevance | path

Searched defs:hidden_state_shape (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc511 TensorShape hidden_state_shape; member
705 const TensorShape& hidden_state_shape = model_shapes.hidden_state_shape; in CreateForwardAndBackwardIODescriptors() local
1591 const TensorShape& hidden_state_shape = model_shapes.hidden_state_shape; in AllocateOutputs() local
2001 const TensorShape& hidden_state_shape = model_shapes.hidden_state_shape; in ExtractBackwardInputs() local
2050 const TensorShape& hidden_state_shape = model_shapes.hidden_state_shape; in AllocateOutputs() local