OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 12:20:24 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[OpenCL] Cleans variable op (#10333)

Browse Source 
* [OpenCL] Cleans variable op

* Fixes formatting and float / double -> GPU_NUMBER_TYPES_NO_HALF
This commit is contained in:
Luke Iwanski 2017-06-07 00:44:40 +01:00 committed by Benoit Steiner
parent 9b2c1af631
commit ed47ecf2dc
1 changed files with 21 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
tensorflow/core/kernels/variable_ops.cc
View File

@ -32,33 +32,29 @@ REGISTER_KERNEL_BUILDER(Name("DestroyTemporaryVariable").Device(DEVICE_CPU),
REGISTER_KERNEL_BUILDER(Name("IsVariableInitialized").Device(DEVICE_CPU),
IsVariableInitializedOp);
#if TENSORFLOW_USE_SYCL
#define REGISTER_SYCL_KERNEL(TYPE) \
REGISTER_KERNEL_BUILDER( \
Name("Variable") \
.Device(DEVICE_SYCL) \
.TypeConstraint<TYPE>("dtype"), \
VariableOp); \
REGISTER_KERNEL_BUILDER(Name("VariableV2") \
.Device(DEVICE_SYCL) \
.TypeConstraint<TYPE>("dtype"), \
VariableOp); \
REGISTER_KERNEL_BUILDER(Name("TemporaryVariable") \
.Device(DEVICE_SYCL) \
.TypeConstraint<TYPE>("dtype"), \
TemporaryVariableOp); \
REGISTER_KERNEL_BUILDER(Name("DestroyTemporaryVariable") \
.Device(DEVICE_SYCL) \
.TypeConstraint<TYPE>("T"), \
DestroyTemporaryVariableOp); \
REGISTER_KERNEL_BUILDER(Name("IsVariableInitialized") \
.Device(DEVICE_SYCL) \
.TypeConstraint<TYPE>("dtype") \
.HostMemory("is_initialized"), \
#ifdef TENSORFLOW_USE_SYCL
#define REGISTER_SYCL_KERNEL(type) \
REGISTER_KERNEL_BUILDER( \
Name("Variable").Device(DEVICE_SYCL).TypeConstraint<type>("dtype"), \
VariableOp); \
REGISTER_KERNEL_BUILDER( \
Name("VariableV2").Device(DEVICE_SYCL).TypeConstraint<type>("dtype"),\
VariableOp); \
REGISTER_KERNEL_BUILDER(Name("TemporaryVariable") \
.Device(DEVICE_SYCL) \
.TypeConstraint<type>("dtype"), \
TemporaryVariableOp); \
REGISTER_KERNEL_BUILDER(Name("DestroyTemporaryVariable") \
.Device(DEVICE_SYCL) \
.TypeConstraint<type>("T"), \
DestroyTemporaryVariableOp); \
REGISTER_KERNEL_BUILDER(Name("IsVariableInitialized") \
.Device(DEVICE_SYCL) \
.TypeConstraint<type>("dtype") \
.HostMemory("is_initialized"), \
IsVariableInitializedOp);
REGISTER_SYCL_KERNEL(float);
REGISTER_SYCL_KERNEL(double);
TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL_KERNEL);
#undef REGISTER_SYCL_KERNEL
#endif // TENSORFLOW_USE_SYCL