def _optimizer_state_shapes(self):
    'Reads the shapes of the optimizer parameters (mutable state).'
    interpreter = tf.lite.Interpreter(model_content=self.optimizer_model)
    num_variables = len(self.variables)
    optim_state_inputs = interpreter.get_input_details()[(num_variables * 2):]
    return [input_['shape'] for input_ in optim_state_inputs]
Reads the shapes of the optimizer parameters (mutable state).
def _optimizer_state_shapes(self):
    
    interpreter = tf.lite.Interpreter(model_content=self.optimizer_model)
    num_variables = len(self.variables)
    optim_state_inputs = interpreter.get_input_details()[(num_variables * 2):]
    return [input_['shape'] for input_ in optim_state_inputs]