Support quantizing atrous convolutions.
Atrous convolutions are often DepthwiseConv2d operations preceded by SpaceToBatchND and followed by BatchToSpaceND operations. This change makes fold_batch_norms.py and quantize.py support handling this pattern. PiperOrigin-RevId: 202353838
Loading
Please sign in to comment