Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/integration_test/
Dtpu_memory_test.py25 NUM_CLASS = 10 variable
37 label = tf.one_hot(label, NUM_CLASS)
38 label = tf.reshape(label, [NUM_CLASS])
132 NUM_CLASS,