QuantizedRnnTanhCell

Applies a quantized RNN cell with Tanh non-linearity to inputs.

Abstract Signature:

QuantizedRnnTanhCell(input: Tensor, hx: Tensor, w_ih: Tensor, w_hh: Tensor, b_ih: Tensor, b_hh: Tensor, packed_ih: Tensor, packed_hh: Tensor, col_offsets: Tensor, scale: float, zero_point: int)

PyTorchJAX (Core)KerasTensorFlowApple MLXFlax NNX

PyTorch

API: torch.quantized_rnn_tanh_cell

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Custom / Partial

Keras

API: —

Strategy: Custom / Partial

TensorFlow

API: —

Strategy: Custom / Partial

Apple MLX

API: —

Strategy: Custom / Partial

Flax NNX

API: —

Strategy: Custom / Partial