`neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction`

Dilated Contraction Graph Rewriter.

Module Contents

Classes

DilatedContraction

Fuse the SpaceToBatchND + Conv + BatchToSpaceND pattern.

class neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction.DilatedContraction(model)[source]: Fuse the SpaceToBatchND + Conv + BatchToSpaceND pattern.