Home
last modified time | relevance | path

Searched refs:CustomModel (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/keras/tests/
Dautomatic_outside_compilation_test.py107 class CustomModel(training.Model): class
111 super(CustomModel, self).__init__()
219 model = CustomModel()
/external/tensorflow/tensorflow/python/keras/distribute/
Dcustom_training_loop_models_test.py43 class CustomModel(module.Module): class
46 super(CustomModel, self).__init__(name=name)
372 model = CustomModel()