rktensor has utilities for converting to flat tensor representations for use with rknpu2.
The main functions are to_tensor and to_tensor_with_quant.
rktensor has utilities for converting to flat tensor representations for use with rknpu2.
The main functions are to_tensor and to_tensor_with_quant.