neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_reshape_transpose

Fuse Transpose and Reshape Graph Rewriter.

Module Contents

Classes

FuseTransposeReshapeOptimizer

Fuse Transpose + Reshape + MatMul/Conv ==> MatMul/Conv.

class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fuse_reshape_transpose.FuseTransposeReshapeOptimizer(model)[source]

Fuse Transpose + Reshape + MatMul/Conv ==> MatMul/Conv.