Saving and loading model in tensorflow 2.0.1

Question

class SimpleDense(Layer):

def __init__(self, units=1, name=\'SimpleDense\',  **kwargs):
    \'\'\'Initializes the instance attributes\'\'\'
    super(SimpleDense, self)