neural_compressor.tensorflow.quantization.utils.graph_rewriter

Subpackages

• neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.bf16_convert
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.bf16.dequantize_cast_optimizer
• neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_add_to_biasadd
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_layout
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_leakyrelu
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_nan_to_random
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.convert_placeholder_to_const
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dilated_contraction
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.dummy_biasadd
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.expanddims_optimizer
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fetch_weight_from_reshape
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_batch_norm
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fold_constant
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_biasadd_add
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_column_wise_mul
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_conv_with_math
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_bn
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_decomposed_in
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_gelu
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_layer_norm
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_conv
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_pad_with_fp32_conv
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.fuse_reshape_transpose
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.graph_cse_optimizer
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.grappler_pass
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.insert_print_node
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.move_squeeze_after_relu
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.pre_optimize
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.remove_training_nodes
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.rename_batch_norm
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.split_shared_input
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_equivalent_nodes
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.strip_unused_nodes
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.generic.switch_optimizer
• neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_fake_quant
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.freeze_value_without_calib
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_redundant_dequantize
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_conv_requantize
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_redundant_dequantize
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.fuse_matmul_requantize
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.meta_op_optimizer
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_hostconst_converter
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.post_quantized_op_cse
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.rnn_convert
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.int8.scale_propagation
• neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.insert_qdq_pattern
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.merge_duplicated_qdq
  • neural_compressor.tensorflow.quantization.utils.graph_rewriter.qdq.share_qdq_y_pattern

Submodules

• neural_compressor.tensorflow.quantization.utils.graph_rewriter.graph_base