Searched defs:use_gpu (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/cc/tutorials/ |
| H A D | example_trainer.cc | 43 bool use_gpu = false; // Whether to use gpu in the training member in struct:tensorflow::example::Options
104 graph::SetDefaultDevice(opts->use_gpu ? "/device:GPU:0" : "/cpu:0", &def);
216 ParseBoolFlag(argv[i], "--use_gpu", &opts.use_gpu)) {
|
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | nn_ops_test.cc | 108 Padding padding, bool use_gpu, DataType data_type,
110 if (!IsGoogleCudaEnabled() && use_gpu) {
221 string device = use_gpu ? "gpu" : "cpu";
503 bool use_gpu, const string& label) {
504 if (!IsGoogleCudaEnabled() && use_gpu) {
601 string device = use_gpu ? "gpu" : "cpu";
1103 bool use_gpu, const string& label) {
1131 string device = use_gpu ? "gpu" : "cpu";
1258 int num_threads, bool use_gpu,
1270 string device = use_gpu
105 BM_ConvFloat(int iters, int batch, int rows, int cols, int in_depth, int out_depth, int filter_rows, int filter_cols, CONV_OP op, int num_threads, int stride, Padding padding, bool use_gpu, DataType data_type, const string& label) argument
498 BM_ConvFloatDepthwise(int iters, int batch, int rows, int cols, int in_depth, int depth_multiplier, int out_depth, int filter_rows, int filter_cols, DEPTHWISE_CONV_OP op, int num_threads, int stride, Padding padding, bool use_gpu, const string& label) argument
1100 BM_MaxPoolBk(int iters, int batch_size, int rows, int cols, int depth, int kernel_rows, int kernel_cols, int stride, Padding padding, int num_threads, bool use_gpu, const string& label) argument
1257 BM_ImageNetSoftmaxFwd(int iters, int batch_size, int node_depth, int num_threads, bool use_gpu, const string& label) argument
1302 BM_TopK(int iters, int rows, int cols, int k, int num_threads, bool use_gpu, const string& label) argument
[all...] |
Completed in 151 milliseconds