experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Register GPU RefExit kernel (#10174)

Browse Source
This commit is contained in:
Daniel Rasmussen 2017-06-15 14:21:32 -04:00 committed by gunan
parent 0d2f691832
commit 7658fdbb90
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/control_flow_ops.cc
View File

@ -447,7 +447,9 @@ REGISTER_KERNEL_BUILDER(Name("RefExit").Device(DEVICE_CPU), ExitOp);
Name("RefExit").Device(DEVICE_GPU).TypeConstraint<type>("T"), ExitOp); Name("RefExit").Device(DEVICE_GPU).TypeConstraint<type>("T"), ExitOp);
TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL); TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL);
TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_REF_KERNEL);
REGISTER_GPU_KERNEL(bool); REGISTER_GPU_KERNEL(bool);
REGISTER_GPU_REF_KERNEL(bool);
#undef REGISTER_GPU_KERNEL #undef REGISTER_GPU_KERNEL
#undef REGISTER_GPU_REF_KERNEL #undef REGISTER_GPU_REF_KERNEL