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

Remove training nodes Graph Rewriter.

Module Contents

Classes

RemoveTrainingNodesOptimizer

Remove training nodes optimizer.

class neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.remove_training_nodes.RemoveTrainingNodesOptimizer(model, protected_nodes=[], types_to_splice=['Identity', 'CheckNumerics', 'StopGradient'])[source]: Remove training nodes optimizer.