Searched defs:BATCH_SIZE (Results 1 – 15 of 15) sorted by relevance
10 const BATCH_SIZE: u32 = 1_000_000; in test_exhaustive() constant
30 BATCH_SIZE = 3 variable in RangeSamplerOpsTest
42 #define BATCH_SIZE 4096 macro
57 #define BATCH_SIZE (1024*1024) macro
44 #define BATCH_SIZE (1024*1024) macro
41 #define BATCH_SIZE 4096 macro
29 #define BATCH_SIZE (4096<<10) macro
16 #define BATCH_SIZE 8 macro
90 private const val BATCH_SIZE = 100000 constant
44 const val BATCH_SIZE = 32 * 8192 constant
65 static const int BATCH_SIZE = 4096; variable
38 #define BATCH_SIZE 64 macro
1409 #define BM_ImageNetSoftmaxFwd(BATCH_SIZE, NODE_DEPTH, TH, GPU, LABEL) \ argument
1501 BATCH_SIZE = 16 variable in UnrollLSTMTest