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

Switch Graph Rewriter.

Module Contents

Classes

SwitchOptimizer

Remove switch op if the input condition is true.

class neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.switch_optimizer.SwitchOptimizer(model)[source]: Remove switch op if the input condition is true.