Searched refs:pvq_quantize (Results 1 – 4 of 4) sorted by relevance
59 from rdovae import pvq_quantize101 state = pvq_quantize(state, 82)
61 from rdovae import pvq_quantize118 gru_state_dec = pvq_quantize(gru_state_dec, 82)
38 from rdovae import new_rdovae_model, pvq_quantize, apply_dead_zone, sq_rate_metric156 quant_gru_state_dec = pvq_quantize(gru_state_dec, 82)
189 def pvq_quantize(x, k): function344 gru_state_dec = Lambda(lambda x: pvq_quantize(x, 82))(gru_state_dec)