quark.onnx.tools.save_tensor_hist#

Module Contents#

Classes#

class quark.onnx.tools.save_tensor_hist.HistDataReader(model_path: str, data_path: str, input_shape: List[int])#

A CalibrationDataReader using random data for rapid quantiation.

get_next() Optional[Dict[str, numpy.typing.NDArray[numpy.float32]]]#

Get next feed data :return: feed dict for the model