def benchmark_keras_model_subclassed_fit(self):
    model = SubclassedKerasModel(initializer='glorot_uniform')
    self._benchmark_keras_model_fit(model)
def benchmark_keras_model_subclassed_fit(self):
    model = SubclassedKerasModel(initializer='glorot_uniform')
    self._benchmark_keras_model_fit(model)