doc: rnn: limit backward layout support for cpu

oneapi-src · Apr 11, 2024 · f81a2a8 · f81a2a8
1 parent 6034b36
commit f81a2a8
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/doc/primitives/rnn.md b/doc/primitives/rnn.md
@@ -431,11 +431,11 @@ primitive parameters.
 The following table summarizes the data layouts supported by the RNN
 primitive.
 
-Propagation        | Input/Output Data    | Recurrent Data       | Layer and Iteration Weights | Peephole Weights and Bias | Projection LSTM Weights
--------------------|----------------------|----------------------|-----------------------------|---------------------------|------------------------
-Forward / Backward | #dnnl_format_tag_any | #dnnl_format_tag_any | #dnnl_format_tag_any        | #dnnl_ldgo                | #dnnl_format_tag_any
-Forward            | #dnnl_ntc, #dnnl_tnc | #dnnl_ldnc           | #dnnl_ldigo                 | #dnnl_ldgo                | #dnnl_ldio
-Backward           | #dnnl_ntc, #dnnl_tnc | #dnnl_ldnc           | #dnnl_ldgoi                 | #dnnl_ldgo                | #dnnl_ldoi
+Propagation        | Input/Output Data    | Recurrent Data       | Layer and Iteration Weights   | Peephole Weights and Bias | Projection LSTM Weights
+-------------------|----------------------|----------------------|-------------------------------|---------------------------|------------------------
+Forward / Backward | #dnnl_format_tag_any | #dnnl_format_tag_any | #dnnl_format_tag_any          | #dnnl_ldgo                | #dnnl_format_tag_any
+Forward            | #dnnl_ntc, #dnnl_tnc | #dnnl_ldnc           | #dnnl_ldigo                   | #dnnl_ldgo                | #dnnl_ldio
+Backward           | #dnnl_ntc, #dnnl_tnc | #dnnl_ldnc           | #dnnl_ldigo, #dnnl_ldgoi(gpu) | #dnnl_ldgo                | #dnnl_ldoi
 
 While an RNN primitive can be created with memory formats specified
 explicitly, the performance is likely to be sub-optimal.  When using `any`, it