QuantizedBatchNorm

Applies batch normalization on a quantized tensor.

Abstract Signature:

QuantizedBatchNorm(input: Tensor, weight: Tensor, bias: Tensor, mean: Tensor, var: Tensor, eps: float, output_scale: float, output_zero_point: int)

PyTorchJAX (Core)KerasTensorFlowApple MLXFlax NNX

PyTorch

API: torch.quantized_batch_norm

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