diff options
Diffstat (limited to 'callbacks/configs.h')
-rw-r--r-- | callbacks/configs.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/callbacks/configs.h b/callbacks/configs.h index 4a01e2e..8e2eacc 100644 --- a/callbacks/configs.h +++ b/callbacks/configs.h @@ -20,6 +20,13 @@ struct UnquantizeAndWrite { UnquantizeAndWrite(float unquant_mult, float* addr) : unquant_mult(unquant_mult), addr(addr) {} }; +struct AddBiasAndWrite { + const int* bias_addr; + int* output_addr; + + AddBiasAndWrite(const int* bias_addr, int* output_addr) : bias_addr(bias_addr), output_addr(output_addr) {} +}; + struct UnquantizeAndAddBiasAndWrite { float unquant_mult; const float* bias_addr; |