Imported from https://github.com/cc-hpc-itwm/TensorQuant