Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/distribute/python/
Dkeras_test.py48 _NUM_CLASS = 2 variable
60 model.add(keras.layers.Dense(_NUM_CLASS, activation='softmax'))
68 b = keras.layers.Dense(_NUM_CLASS, activation='softmax')(b)
73 def simple_subclassed_model(num_labels=_NUM_CLASS):
131 num_classes=_NUM_CLASS)
145 num_classes=_NUM_CLASS)
Dkeras_backward_compat_test.py43 _NUM_CLASS = 2 variable
52 model.add(keras.layers.Dense(_NUM_CLASS, activation='softmax'))
60 b = keras.layers.Dense(_NUM_CLASS, activation='softmax')(b)
97 num_classes=_NUM_CLASS)
111 num_classes=_NUM_CLASS)