Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/tests/
Dadd_loss_correctness_test.py338 class LayerWithLoss(layers.Layer): class
344 shared_layer = LayerWithLoss()
355 class LayerWithLoss(layers.Layer): class
365 self.loss_layer = LayerWithLoss()
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dsaved_model_test.py91 class LayerWithLoss(keras.layers.Layer): class
213 model = keras.models.Sequential([LayerWithLoss()])
999 layer = LayerWithLoss()