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

Add missing "CPU" suffix in registrations.

Browse Source 
PiperOrigin-RevId: 164939527
This commit is contained in:
A. Unique TensorFlower 2017-08-10 20:45:03 -07:00 committed by TensorFlower Gardener
parent e2a163a905
commit eeacdcdb14
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/dynamic_stitch_op.cc
View File

@ -360,14 +360,14 @@ TF_CALL_int32(REGISTER_DYNAMIC_STITCH_GPU);
.HostMemory("indices") \
.HostMemory("data") \
.HostMemory("merged"), \
DynamicStitchOp<type>) \
DynamicStitchOpCPU<type>) \
REGISTER_KERNEL_BUILDER(Name("ParallelDynamicStitch") \
.Device(DEVICE_SYCL) \
.TypeConstraint<type>("T") \
.HostMemory("indices") \
.HostMemory("data") \
.HostMemory("merged"), \
ParallelDynamicStitchOp<type>)
ParallelDynamicStitchOpCPU<type>)
TF_CALL_POD_STRING_TYPES(REGISTER_DYNAMIC_STITCH_SYCL);
#undef REGISTER_DYNAMIC_STITCH_SYCL