/external/tensorflow/tensorflow/python/keras/_impl/keras/ |
H A D | callbacks_test.py | 51 BATCH_SIZE = 5 variable 99 batch_size=BATCH_SIZE, 119 batch_size=BATCH_SIZE, 140 batch_size=BATCH_SIZE, 160 batch_size=BATCH_SIZE, 178 batch_size=BATCH_SIZE, 203 batch_size=BATCH_SIZE, 258 batch_size=BATCH_SIZE, 318 batch_size=BATCH_SIZE, 364 batch_size=BATCH_SIZE, [all...] |
/external/tensorflow/tensorflow/python/keras/_impl/keras/wrappers/ |
H A D | scikit_learn_test.py | 32 BATCH_SIZE = 5 variable 57 clf.fit(x_train, y_train, batch_size=BATCH_SIZE, epochs=EPOCHS) 59 score = clf.score(x_train, y_train, batch_size=BATCH_SIZE) 62 preds = clf.predict(x_test, batch_size=BATCH_SIZE) 67 proba = clf.predict_proba(x_test, batch_size=BATCH_SIZE) 93 reg.fit(x_train, y_train, batch_size=BATCH_SIZE, epochs=EPOCHS) 95 score = reg.score(x_train, y_train, batch_size=BATCH_SIZE) 98 preds = reg.predict(x_test, batch_size=BATCH_SIZE) 109 batch_size=BATCH_SIZE, 125 batch_size=BATCH_SIZE, [all...] |
/external/tensorflow/tensorflow/python/kernel_tests/ |
H A D | candidate_sampler_ops_test.py | 33 BATCH_SIZE = 3 variable in class:RangeSamplerOpsTest 45 true_candidates_vec, [self.BATCH_SIZE, self.NUM_TRUE]) 73 self.assertAllEqual(result, [[0.0] * self.NUM_TRUE] * self.BATCH_SIZE) 75 [self.BATCH_SIZE, self.NUM_TRUE]) 77 [self.BATCH_SIZE, self.NUM_TRUE])
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/ |
H A D | rnn_ptb_graph_test.py | 62 BATCH_SIZE = 20 variable in class:PTBBenchmark 83 [PTBBenchmark.SEQ_LEN, PTBBenchmark.BATCH_SIZE], 100 tf.test.gpu_device_name(), PTBBenchmark.BATCH_SIZE) 123 [PTBBenchmark.SEQ_LEN, PTBBenchmark.BATCH_SIZE], 144 tf.test.gpu_device_name(), PTBBenchmark.BATCH_SIZE)
|
H A D | rnn_ptb_test.py | 60 BATCH_SIZE = 20 variable in class:PTBBenchmark 79 [PTBBenchmark.SEQ_LEN, PTBBenchmark.BATCH_SIZE], dtype=tf.int64) 117 [PTBBenchmark.SEQ_LEN, PTBBenchmark.BATCH_SIZE], dtype=tf.int64)
|
/external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/ |
H A D | cifar10_pruning.py | 53 BATCH_SIZE = 128 variable 145 data_dir=data_dir, batch_size=BATCH_SIZE) 166 eval_data=eval_data, data_dir=data_dir, batch_size=BATCH_SIZE) 235 reshape = tf.reshape(pool2, [BATCH_SIZE, -1]) 338 num_batches_per_epoch = NUM_EXAMPLES_PER_EPOCH_FOR_TRAIN / BATCH_SIZE
|
/external/tensorflow/tensorflow/contrib/layers/python/layers/ |
H A D | rev_block_lib_test.py | 41 BATCH_SIZE = 16 variable in class:RevBlockTest 52 [self.BATCH_SIZE, self.CHANNELS], dtype=dtypes.float32) 75 [self.BATCH_SIZE, self.CHANNELS], dtype=dtypes.float32) 115 [self.BATCH_SIZE, self.CHANNELS], dtype=dtypes.float32) 164 [self.BATCH_SIZE, self.CHANNELS // 2]) 187 [self.BATCH_SIZE, 10, self.CHANNELS], dtype=dtypes.float32) 207 [self.BATCH_SIZE, self.CHANNELS], dtype=dtypes.float32)
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | nn_ops_test.cc | 1284 #define BM_ImageNetSoftmaxFwd(BATCH_SIZE, NODE_DEPTH, TH, GPU, LABEL) \ 1286 BM_ImageNetSoftmaxFwd_##BATCH_SIZE##_##NODE_DEPTH##_##TH##_##GPU( \ 1288 BM_ImageNetSoftmaxFwd(iters, BATCH_SIZE, NODE_DEPTH, TH, GPU, LABEL); \ 1290 BENCHMARK(BM_ImageNetSoftmaxFwd_##BATCH_SIZE##_##NODE_DEPTH##_##TH##_##GPU)
|
/external/tensorflow/tensorflow/python/framework/ |
H A D | function_test.py | 1309 BATCH_SIZE = 16 variable in class:UnrollLSTMTest 1319 [self.NUM_UNROLL, self.BATCH_SIZE, self.LSTM_DIMS], seed=654321)
|