neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random

Convert NAN to random Graph Rewriter.

Module Contents

Classes

ConvertNanToRandom

Convert Const node which value consists of NAN to random data.

class neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random.ConvertNanToRandom(model)[source]

Convert Const node which value consists of NAN to random data.