neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.dequantize_cast_optimizer
Dequantize Cast Graph Rerewriter.
Module Contents
Classes
Remove the Cast OP and set Dequantize output to B16 if the Cast OP output is BF16. |