-
-
Notifications
You must be signed in to change notification settings - Fork 751
Description
Tensorrt offers class IInt8EntropyCalibrator2 for providing scale calibration data (batches) for model conversion to 8-bit.
This class has 4 pure virtual methods getBatchSize, getBatch, readCalibrationCache and writeCalibrationCache; Purpose of class is to allow custom implementation of these methods.
In order to be useful, class org.bytedeco.tensorrt.nvinfer.IInt8EntropyCalibrator2 should be possible to extend and implement above methods in Java.
To me it seems doing this is quite difficult. The constructor public IInt8EntropyCalibrator2(Pointer p) expects native object, so extending still requires use of javacpp (not just pure java).
Would it be possible to have org.bytedeco.tensorrt.nvinfer.IInt8EntropyCalibrator2 in such way that it has contructor public IInt8EntropyCalibrator2() and abstract java methods getBatchSize and getBatch; and native code would route any c++ calls to these methods into Java?