`neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fetch_weight_from_reshape`

Fetch Weight from Reshape Graph Rewriter.

Module Contents

Classes

FetchWeightFromReshapeOptimizer

Handle the Pack + Reshape + Conv2D fusion pattern.

class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fetch_weight_from_reshape.FetchWeightFromReshapeOptimizer(model)[source]: Handle the Pack + Reshape + Conv2D fusion pattern.