DumpFakeQuantFM.DumpFakeQuantFM Class Reference

Public Member Functions
	StartAnalysis (self, args)
	EndNetworkExecution (self, list outputs)
	DefaultOpPost (self, name, opcode, inputs, outputs)
	EndAnalysis (self)

Protected Attributes
	_dir
	_num_data
	_tname_to_tid
	_scale_map

Detailed Description

Definition at line 49 of file DumpFakeQuantFM.py.

Member Function Documentation

DefaultOpPost()

DumpFakeQuantFM.DumpFakeQuantFM.DefaultOpPost	(		self,
			name,
			opcode,
			inputs,
			outputs
	)

Definition at line 61 of file DumpFakeQuantFM.py.

    def DefaultOpPost(self, name, opcode, inputs, outputs):
        if opcode == 'Dequantize':
            for output in outputs:
                name = output['name']
                data = output['data']
                orig_name = _name_before_fq(name)
                if orig_name in self._tname_to_tid:
                    tid = self._tname_to_tid[orig_name]
                    data_path = self._dir / str(self._num_data)
                    data_path.mkdir(parents=False, exist_ok=True)
                    np.save(str(data_path / str(tid)), data)
                    # Save scales (scale is fixed, so saving once)
                    if orig_name not in self._scale_map:
                        assert len(inputs) == 1
                        assert 'quantparam' in inputs[0]
                        assert 'scale' in inputs[0]['quantparam']
                        assert len(inputs[0]['quantparam']['scale']) == 1
                        scale = inputs[0]['quantparam']['scale'][0]
                        self._scale_map[orig_name] = scale
 

References DumpFakeQuantFM.DumpFakeQuantFM._dir, DumpFP32FM.DumpFP32FM._dir, DumpFakeQuantFM._name_before_fq(), circle_eval_diff::MAEPrinter._num_data, circle_eval_diff::MSEPrinter._num_data, circle_eval_diff::MAPEPrinter._num_data, circle_eval_diff::MPEIRPrinter._num_data, circle_eval_diff::TopKMatchPrinter._num_data, record_minmax::HDF5Iterator._num_data, record_minmax::RandomIterator._num_data, DumpFakeQuantFM.DumpFakeQuantFM._num_data, DumpFP32FM.DumpFP32FM._num_data, QErrorComputer.QErrorComputer._num_data, DumpFakeQuantFM.DumpFakeQuantFM._scale_map, DumpFakeQuantFM.DumpFakeQuantFM._tname_to_tid, and DumpFP32FM.DumpFP32FM._tname_to_tid.

EndAnalysis()

DumpFakeQuantFM.DumpFakeQuantFM.EndAnalysis

(

self

)

Definition at line 81 of file DumpFakeQuantFM.py.

    def EndAnalysis(self):
        # Dump saved scales into scales.txt
        with open(self._dir / 'scales.txt', 'w') as f:
            json.dump(self._scale_map, f)

References DumpFakeQuantFM.DumpFakeQuantFM._dir, DumpFP32FM.DumpFP32FM._dir, and DumpFakeQuantFM.DumpFakeQuantFM._scale_map.

EndNetworkExecution()

DumpFakeQuantFM.DumpFakeQuantFM.EndNetworkExecution	(		self,
		list	outputs
	)

Definition at line 57 of file DumpFakeQuantFM.py.

    def EndNetworkExecution(self, outputs: list):
        self._num_data += 1
 

References circle_eval_diff::MAEPrinter._num_data, circle_eval_diff::MSEPrinter._num_data, circle_eval_diff::MAPEPrinter._num_data, circle_eval_diff::MPEIRPrinter._num_data, circle_eval_diff::TopKMatchPrinter._num_data, record_minmax::HDF5Iterator._num_data, record_minmax::RandomIterator._num_data, DumpFakeQuantFM.DumpFakeQuantFM._num_data, DumpFP32FM.DumpFP32FM._num_data, and QErrorComputer.QErrorComputer._num_data.

StartAnalysis()

DumpFakeQuantFM.DumpFakeQuantFM.StartAnalysis	(		self,
			args
	)

Definition at line 50 of file DumpFakeQuantFM.py.

    def StartAnalysis(self, args):
        self._dir = Path(args)
        self._num_data = 0
        with open(self._dir / 'tensors.json') as f:
            self._tname_to_tid = json.load(f)
        self._scale_map = {}
 

Field Documentation

_dir

DumpFakeQuantFM.DumpFakeQuantFM._dir

protected

Definition at line 51 of file DumpFakeQuantFM.py.

Referenced by DumpFakeQuantFM.DumpFakeQuantFM.DefaultOpPost(), DumpFP32FM.DumpFP32FM.DefaultOpPost(), DumpFakeQuantFM.DumpFakeQuantFM.EndAnalysis(), and DumpFP32FM.DumpFP32FM.EndAnalysis().

_num_data

DumpFakeQuantFM.DumpFakeQuantFM._num_data

protected

Definition at line 52 of file DumpFakeQuantFM.py.

Referenced by DumpFakeQuantFM.DumpFakeQuantFM.DefaultOpPost(), DumpFP32FM.DumpFP32FM.DefaultOpPost(), DumpFakeQuantFM.DumpFakeQuantFM.EndNetworkExecution(), and DumpFP32FM.DumpFP32FM.EndNetworkExecution().

_scale_map

DumpFakeQuantFM.DumpFakeQuantFM._scale_map

protected

Definition at line 55 of file DumpFakeQuantFM.py.

Referenced by DumpFakeQuantFM.DumpFakeQuantFM.DefaultOpPost(), and DumpFakeQuantFM.DumpFakeQuantFM.EndAnalysis().

_tname_to_tid

DumpFakeQuantFM.DumpFakeQuantFM._tname_to_tid

protected

Definition at line 54 of file DumpFakeQuantFM.py.

Referenced by DumpFakeQuantFM.DumpFakeQuantFM.DefaultOpPost(), DumpFP32FM.DumpFP32FM.DefaultOpPost(), and DumpFP32FM.DumpFP32FM.EndAnalysis().

---

The documentation for this class was generated from the following file:

• compiler/visq/visqlib/DumpFakeQuantFM.py