keras_rt.py¶
-
class
keras_rt.
KerasRT
(keras_model, xclbin_opts, wgt_scale, bias_scale, post_scale)¶ runtime class to use FCN in the keras example
It will send all the weights matrices loaded from the model to kernel for future usage
- Parameters
xclbin_opts (dictionary) – information read from config_info.dat used to build the xclbin
keras_model (model) – pre-trained model
wgt_scale (list) – Quantization parameters multiple with weight matrices
bias_scale (list) – Quantization parameters multiple with bias matrices
post_scale (list) – Quantization parameters multiple with output matrices