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)`` .. raw:: html

PyTorch

API: torch.quantized_batch_norm
Strategy: Direct Mapping

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