| /external/tensorflow/tensorflow/tools/android/test/src/org/tensorflow/demo/ |
| D | TensorFlowYoloDetector.java | 37 private static final int NUM_CLASSES = 20; field in TensorFlowYoloDetector
171 new float[gridWidth * gridHeight * (NUM_CLASSES + 5) * NUM_BOXES_PER_BLOCK]; in recognizeImage()
191 (gridWidth * (NUM_BOXES_PER_BLOCK * (NUM_CLASSES + 5))) * y in recognizeImage()
192 + (NUM_BOXES_PER_BLOCK * (NUM_CLASSES + 5)) * x in recognizeImage()
193 + (NUM_CLASSES + 5) * b; in recognizeImage()
212 final float[] classes = new float[NUM_CLASSES]; in recognizeImage()
213 for (int c = 0; c < NUM_CLASSES; ++c) { in recognizeImage()
218 for (int c = 0; c < NUM_CLASSES; ++c) { in recognizeImage()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | regularizers_test.py | 35 NUM_CLASSES = 2 variable
43 model.add(keras.layers.Dense(NUM_CLASSES,
54 num_classes=NUM_CLASSES)
55 y_train = np_utils.to_categorical(y_train, NUM_CLASSES)
56 y_test = np_utils.to_categorical(y_test, NUM_CLASSES)
137 NUM_CLASSES,
155 NUM_CLASSES,
177 NUM_CLASSES,
184 NUM_CLASSES, kernel_regularizer=regularizer)
|
| D | callbacks_v1_test.py | 45 NUM_CLASSES = 2 variable
63 num_classes=NUM_CLASSES)
91 model.add(layers.Dense(NUM_CLASSES, activation='softmax'))
171 num_classes=NUM_CLASSES)
197 output1 = layers.Dense(NUM_CLASSES, activation='softmax')(hidden)
198 output2 = layers.Dense(NUM_CLASSES, activation='softmax')(hidden)
274 num_classes=NUM_CLASSES)
285 model.add(layers.Dense(NUM_CLASSES, activation='softmax'))
369 num_classes=NUM_CLASSES)
374 num_hidden=NUM_HIDDEN, num_classes=NUM_CLASSES, input_dim=INPUT_DIM)
[all …]
|
| D | callbacks_test.py | 76 NUM_CLASSES = 2 variable
453 keras.layers.Dense(NUM_CLASSES, activation='softmax')
467 num_classes=NUM_CLASSES)
479 model.add(keras.layers.Dense(NUM_CLASSES, activation='softmax'))
1014 num_classes=NUM_CLASSES)
1018 num_hidden=NUM_HIDDEN, num_classes=NUM_CLASSES, input_dim=INPUT_DIM)
1072 num_classes=NUM_CLASSES)
1161 num_classes=NUM_CLASSES)
1165 num_hidden=NUM_HIDDEN, num_classes=NUM_CLASSES, input_dim=INPUT_DIM)
1232 num_classes=NUM_CLASSES)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/wrappers/ |
| D | scikit_learn_test.py | 32 NUM_CLASSES = 2 variable
43 model.add(keras.layers.Dense(NUM_CLASSES))
56 num_classes=NUM_CLASSES)
66 assert prediction in range(NUM_CLASSES)
69 assert proba.shape == (TEST_SAMPLES, NUM_CLASSES)
92 num_classes=NUM_CLASSES)
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/integration_test/benchmarks/ |
| D | numpy_mlp.py | 22 NUM_CLASSES = 3 variable
34 def __init__(self, num_classes=NUM_CLASSES, input_size=INPUT_SIZE,
|
| D | tf_numpy_mlp.py | 24 NUM_CLASSES = 3 variable
36 def __init__(self, num_classes=NUM_CLASSES, input_size=INPUT_SIZE,
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | automatic_outside_compilation_test.py | 54 NUM_CLASSES = 4 variable
136 targets = np.zeros((10, NUM_CLASSES), dtype=np.float32)
159 model.add(layer_lib.Dense(NUM_CLASSES, activation='softmax'))
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/tutorials/ |
| D | jit_compile.ipynb | 117 "NUM_CLASSES = 10\n",
152 "layer = tf.keras.layers.Dense(NUM_CLASSES)\n",
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_Numpy_Distributed_Image_Classification.ipynb | 133 "NUM_CLASSES = 10\n",
141 " labels = tnp.eye(NUM_CLASSES, dtype=tnp.float32)[labels]\n",
239 " self.layer3 = Dense(NUM_CLASSES, use_relu=False)\n",
|