neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input

Split shared input Graph Rewriter.

Module Contents

Classes

SplitSharedInputOptimizer

Split the shared input if the input node is shared and const.

class neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input.SplitSharedInputOptimizer(model)[source]

Split the shared input if the input node is shared and const.