diff options
Diffstat (limited to 'tensorflow/core/framework/types.h')
-rw-r--r-- | tensorflow/core/framework/types.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/core/framework/types.h b/tensorflow/core/framework/types.h index 7d4ed9a004..589730baf1 100644 --- a/tensorflow/core/framework/types.h +++ b/tensorflow/core/framework/types.h @@ -68,8 +68,9 @@ class DeviceType { std::ostream& operator<<(std::ostream& os, const DeviceType& d); // Convenient constants that can be passed to a DeviceType constructor -extern const char* const DEVICE_CPU; // "CPU" -extern const char* const DEVICE_GPU; // "GPU" +extern const char* const DEVICE_CPU; // "CPU" +extern const char* const DEVICE_GPU; // "GPU" +extern const char* const DEVICE_SYCL; // "SYCL" typedef gtl::InlinedVector<MemoryType, 4> MemoryTypeVector; typedef gtl::ArraySlice<MemoryType> MemoryTypeSlice; |