QuantizedGruCell

Applies a quantized GRU cell to inputs.

Abstract Signature:

QuantizedGruCell(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_gru_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