diff options
Diffstat (limited to 'tensorflow/core/graph/mkl_layout_pass_test.cc')
-rw-r--r-- | tensorflow/core/graph/mkl_layout_pass_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/graph/mkl_layout_pass_test.cc b/tensorflow/core/graph/mkl_layout_pass_test.cc index efbe2134e0..482e339802 100644 --- a/tensorflow/core/graph/mkl_layout_pass_test.cc +++ b/tensorflow/core/graph/mkl_layout_pass_test.cc @@ -40,7 +40,7 @@ namespace tensorflow { namespace { const char kCPUDevice[] = "/job:a/replica:0/task:0/cpu:0"; -const char kGPUDevice[] = "/job:a/replica:0/task:0/gpu:0"; +const char kGPUDevice[] = "/job:a/replica:0/task:0/device:GPU:0"; static void InitGraph(const string& s, Graph* graph, const string& device = kCPUDevice) { |