DumpFP32FM.DumpFP32FM Class Reference

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

Protected Attributes
	_dir
	_num_data
	_tname_to_tid
	_tensor_count

Detailed Description

Definition at line 34 of file DumpFP32FM.py.

Member Function Documentation

DefaultOpPost()

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

Definition at line 45 of file DumpFP32FM.py.

    def DefaultOpPost(self, name, opcode, inputs, outputs):
        # Save intermediate FM into <tid>.npy
        data_path = self._dir / str(self._num_data)
        data_path.mkdir(parents=False, exist_ok=True)
        for output in outputs:
            name = output['name']
            data = output['data']
            if name in self._tname_to_tid:
                tid = self._tname_to_tid[name]
            else:
                tid = self._tensor_count
                self._tname_to_tid[name] = tid
                self._tensor_count += 1
 
            np.save(str(data_path / str(tid)), data)
 

References record_minmax::DirectoryIterator._dir, DumpFakeQuantFM.DumpFakeQuantFM._dir, DumpFP32FM.DumpFP32FM._dir, 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, DumpFP32FM.DumpFP32FM._tensor_count, DumpFakeQuantFM.DumpFakeQuantFM._tname_to_tid, and DumpFP32FM.DumpFP32FM._tname_to_tid.

EndAnalysis()

DumpFP32FM.DumpFP32FM.EndAnalysis

(

self

)

Definition at line 61 of file DumpFP32FM.py.

    def EndAnalysis(self):
        # Save tensor name : tensor id pairs
        with open(self._dir / 'tensors.json', 'w') as f:
            json.dump(self._tname_to_tid, f, indent=2)

References record_minmax::DirectoryIterator._dir, DumpFakeQuantFM.DumpFakeQuantFM._dir, DumpFP32FM.DumpFP32FM._dir, DumpFakeQuantFM.DumpFakeQuantFM._tname_to_tid, and DumpFP32FM.DumpFP32FM._tname_to_tid.

EndNetworkExecution()

DumpFP32FM.DumpFP32FM.EndNetworkExecution	(		self,
			outputs
	)

Definition at line 42 of file DumpFP32FM.py.

    def EndNetworkExecution(self, outputs):
        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()

DumpFP32FM.DumpFP32FM.StartAnalysis	(		self,
			args
	)

Definition at line 35 of file DumpFP32FM.py.

    def StartAnalysis(self, args):
        self._dir = Path(args)
        self._num_data = 0
        # Dict {tensor_name -> tid}
        self._tname_to_tid = dict()
        self._tensor_count = 0
 

Field Documentation

_dir

DumpFP32FM.DumpFP32FM._dir

protected

Definition at line 36 of file DumpFP32FM.py.

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

_num_data

DumpFP32FM.DumpFP32FM._num_data

protected

Definition at line 37 of file DumpFP32FM.py.

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

_tensor_count

DumpFP32FM.DumpFP32FM._tensor_count

protected

Definition at line 40 of file DumpFP32FM.py.

Referenced by DumpFP32FM.DumpFP32FM.DefaultOpPost().

_tname_to_tid

DumpFP32FM.DumpFP32FM._tname_to_tid

protected

Definition at line 39 of file DumpFP32FM.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/DumpFP32FM.py