diff options
author | 2017-08-10 20:45:03 -0700 | |
---|---|---|
committer | 2017-08-10 20:48:17 -0700 | |
commit | eeacdcdb14bdd175665a902e02c99e4435d0484f (patch) | |
tree | dc9731a3989e72b5d283b79728fb45b1991118b9 /tensorflow/core/kernels/dynamic_stitch_op.cc | |
parent | e2a163a90561bef0accdd7a0f200f692d85e14c9 (diff) |
Add missing "CPU" suffix in registrations.
PiperOrigin-RevId: 164939527
Diffstat (limited to 'tensorflow/core/kernels/dynamic_stitch_op.cc')
-rw-r--r-- | tensorflow/core/kernels/dynamic_stitch_op.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/dynamic_stitch_op.cc b/tensorflow/core/kernels/dynamic_stitch_op.cc index 99bcd90a4e..f018499f6c 100644 --- a/tensorflow/core/kernels/dynamic_stitch_op.cc +++ b/tensorflow/core/kernels/dynamic_stitch_op.cc @@ -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 |