neural_compressor.adaptor.tf_utils.graph_rewriter.generic.move_squeeze_after_relu

Move Squeeze after Relu Graph Rewriter.

Module Contents

Classes

MoveSqueezeAfterReluOptimizer

Move Squeeze op after Relu op for match fusion pattern.

class neural_compressor.adaptor.tf_utils.graph_rewriter.generic.move_squeeze_after_relu.MoveSqueezeAfterReluOptimizer(model)[source]

Move Squeeze op after Relu op for match fusion pattern.