def benchmark_keras_model_subclassed_predict(self):
    model = SubclassedKerasModel(initializer='glorot_uniform')
    self._benchmark_keras_model_predict(model)
def benchmark_keras_model_subclassed_predict(self):
    model = SubclassedKerasModel(initializer='glorot_uniform')
    self._benchmark_keras_model_predict(model)