def benchmark_keras_model_subclassed_fit_graph_mode(self):
    with context.graph_mode():
        model = SubclassedKerasModel(initializer='glorot_uniform')
        self._benchmark_keras_model_fit(model)
def benchmark_keras_model_subclassed_fit_graph_mode(self):
    with context.graph_mode():
        model = SubclassedKerasModel(initializer='glorot_uniform')
        self._benchmark_keras_model_fit(model)