Lines Matching refs:GPU
112 strings::StrCat("Skipping GPU test (no --config=cuda): ", label));
221 string device = use_gpu ? "gpu" : "cpu";
506 strings::StrCat("Skipping GPU test (no --config=cuda): ", label));
601 string device = use_gpu ? "gpu" : "cpu";
1131 string device = use_gpu ? "gpu" : "cpu";
1270 string device = use_gpu ? "gpu" : "cpu";
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)
1318 string device = use_gpu ? "gpu" : "cpu";
