diff options
| author |  Geoffrey Irving <geoffreyi@google.com> | 2017-05-16 17:01:31 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-16 17:05:15 -0700 |
| commit | 43db5c623f748b6f9704e9e9be5a5a11fa2a4c1a (patch) | |
| tree | 985844ec8f6653f36e38592f9700dcaba66d94f2 /tensorflow/core/kernels/function_ops.cc | |
| parent | 7ab0c2eff12ea79648f6717dae8558d6669e5c27 (diff) | |
Automated g4 rollback of changelist 156244933
PiperOrigin-RevId: 156251356
Diffstat (limited to 'tensorflow/core/kernels/function_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/function_ops.cc | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/tensorflow/core/kernels/function_ops.cc b/tensorflow/core/kernels/function_ops.cc index 8c3137ece9..ba408f3657 100644 --- a/tensorflow/core/kernels/function_ops.cc +++ b/tensorflow/core/kernels/function_ops.cc @@ -186,12 +186,12 @@ REGISTER_KERNEL_BUILDER(Name("_ArrayToList") PassOn); #ifdef TENSORFLOW_USE_SYCL -#define REGISTER_SYCL_KERNELS(type) \ - REGISTER_KERNEL_BUILDER( \ - Name("_ListToArray").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \ - PassOn); \ - REGISTER_KERNEL_BUILDER( \ - Name("_ArrayToList").Device(DEVICE_SYCL).TypeConstraint<type>("T"), \ +#define REGISTER_SYCL_KERNELS(type) \ + REGISTER_KERNEL_BUILDER( \ + Name("_ListToArray").Device(DEVICE_SYCL).TypeConstraint<type>("T"),\ + PassOn); \ + REGISTER_KERNEL_BUILDER( \ + Name("_ArrayToList").Device(DEVICE_SYCL).TypeConstraint<type>("T"),\ PassOn); REGISTER_SYCL_KERNELS(float); @@ -211,7 +211,7 @@ REGISTER_KERNEL_BUILDER(Name("_ArrayToList") .HostMemory("output") .TypeConstraint<int32>("T"), PassOn); -#endif // TENSORFLOW_USE_SYCL +#endif // TENSORFLOW_USE_SYCL class SymbolicGradientOp : public AsyncOpKernel { public: @@ -227,7 +227,7 @@ class SymbolicGradientOp : public AsyncOpKernel { FunctionLibraryRuntime::Handle handle; OP_REQUIRES_OK_ASYNC( - ctx, lib->Instantiate(kGradientOp, AttrSlice(def()), &handle), done); + ctx, lib->Instantiate(kGradientOp, def().attr(), &handle), done); FunctionLibraryRuntime::Options opts; opts.step_id = ctx->step_id(); |