neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base

QuantizeGraph Base Class.

Module Contents

Classes

QuantizeGraphBase

This is the base class for quantize graph.

QuantizeNodeBase

This is the base class for nodes fusion.

class neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base.QuantizeGraphBase(output_node_names)[source]

This is the base class for quantize graph.

class neural_compressor.tensorflow.quantization.utils.quantize_graph.quantize_graph_base.QuantizeNodeBase(**kwargs)[source]

This is the base class for nodes fusion.

Parameters:

[description] (object {[type]} --) –