Searched refs:BATCH_SIZE (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/keras/_impl/keras/
H A Dcallbacks_test.py51 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 Dscikit_learn_test.py32 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 Dcandidate_sampler_ops_test.py33 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 Drnn_ptb_graph_test.py62 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 Drnn_ptb_test.py60 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 Dcifar10_pruning.py53 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 Drev_block_lib_test.py41 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 Dnn_ops_test.cc1284 #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 Dfunction_test.py1309 BATCH_SIZE = 16 variable in class:UnrollLSTMTest
1319 [self.NUM_UNROLL, self.BATCH_SIZE, self.LSTM_DIMS], seed=654321)

Completed in 328 milliseconds