MkldnnRnnLayer

Applies RNN layer (MKLDNN).

Abstract Signature:

MkldnnRnnLayer(input: Tensor, weight0: Tensor, weight1: Tensor)

PyTorchJAX (Core)Keras

PyTorch

API: torch.mkldnn_rnn_layer

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Custom / Partial

Keras

API: keras.layers.SimpleRNN

Strategy: Direct Mapping

Official Docs ↗