`neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert`

Graph rewriter BF16 Converter Class.

Module Contents

Classes

BF16 node convert transformation.

class neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert.BF16Convert(model, fp32_ops=[], bf16_ops=[])[source]: BF16 node convert transformation.