Searched refs:output_state_scale (Results 1 – 1 of 1) sorted by relevance
257 output_state_scale = output_state['quantization_parameters']['scales'][0]267 self.calc_scales(input_scale, output_state_scale, cell_scale)311 def calc_scales(self, input_scale, output_state_scale, cell_scale): argument322 self.effective_hidden_scale = pow(2, -15) / output_state_scale * pow(2, -15)333 …self.r2i_effective_scale = output_state_scale * self.lstm_scales[self.recurrent_input_to_input_w_i…335 …self.r2f_effective_scale = output_state_scale * self.lstm_scales[self.recurrent_input_to_forget_w_…337 …self.r2c_effective_scale = output_state_scale * self.lstm_scales[self.recurrent_input_to_cell_w_in…339 …self.r2o_effective_scale = output_state_scale * self.lstm_scales[self.recurrent_input_to_output_w_…
