Example of dnnl::vanilla_rnn_backward

[dtecta]

Hi all,

I'm exploring the use of oneDNN for character animation (retargeting, IK). I'm coding directly in C++ and I'm looking for and example showing a full implementation of RNN training using oneDNN. The current distribution at https://github.com/uxlfoundation/oneDNN.git shows only the forward pass, dnnl::vanilla_rnn_forward, in vanilla_rnn.cpp.

In particular, I'm interested in a code sample that shows the use of dnnl::vanilla_rnn_backward.

There is a rnn_training_f32.cpp example that shows both a forward and backward pass. However, this example uses LSTM and is rather contrived and I want to start simple.

Could someone please provide me with a code sample showing the use of dnnl::vanilla_rnn_backward in a typical use case? Thank you!

[rupakroyintel]

@dtecta Thanks for reaching out. We will work on this request and keep you posted.

[chaoyu3]

@dtecta, For the example code post by @raistefintel : https://github.com/uxlfoundation/oneDNN/blob/379935a6c5cbe7cf74572a647c7bf51c126be0d6/examples/primitives/vanilla_rnn.cpp
Does that work for you?