Implementation of QuantGRU

[bimsaraperera]

Can I know whether there's a implementation of GRU block in brevitas or any other workaround for that?

[Giuseppe5]

Unfortunately, we only have QuantRNN and QuantLSTM implemented, and you can find them here:
https://github.com/Xilinx/brevitas/blob/master/src/brevitas/nn/quant_rnn.py

You could start from the QuantLSTM implementation and put together the QuantGRU one, but you would need to pay attention to the quantizers and their placement.