Home
last modified time | relevance | path

Searched refs:place_all_on_gpu (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_test.cc594 const bool place_all_on_gpu = allow_gpu_device && has_gpu_device; in RunAndFetch() local
596 const string device = place_all_on_gpu ? "/device:GPU:0" : "/device:CPU:0"; in RunAndFetch()