diff options
| author |  Derek Murray <mrry@google.com> | 2017-04-19 19:45:54 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-04-19 21:08:17 -0700 |
| commit | 91e4a19c1bbded4430299c636486c86b3fff2ab9 (patch) | |
| tree | 7c15cd553d00992ab588326f35b5ae9758d87fda /tensorflow/core/kernels/function_ops.cc | |
| parent | 225f4e323e8f76807e6bbc9c3026dcd37d828fed (diff) | |
Add REGISTER_SYSTEM_OP and REGISTER_SYSTEM_KERNEL_BUILDER macros.
Use these macros to circumvent the selective registration mechanism
and always build the "_Arg" and "_Retval" ops.
Change: 153669926
Diffstat (limited to 'tensorflow/core/kernels/function_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/function_ops.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/function_ops.cc b/tensorflow/core/kernels/function_ops.cc index bcbf7424b1..ba408f3657 100644 --- a/tensorflow/core/kernels/function_ops.cc +++ b/tensorflow/core/kernels/function_ops.cc @@ -84,8 +84,8 @@ class RetvalOp : public OpKernel { TF_DISALLOW_COPY_AND_ASSIGN(RetvalOp); }; -REGISTER_KERNEL_BUILDER(Name("_Arg").Device(DEVICE_CPU), ArgOp); -REGISTER_KERNEL_BUILDER(Name("_Retval").Device(DEVICE_CPU), RetvalOp); +REGISTER_SYSTEM_KERNEL_BUILDER(Name("_Arg").Device(DEVICE_CPU), ArgOp); +REGISTER_SYSTEM_KERNEL_BUILDER(Name("_Retval").Device(DEVICE_CPU), RetvalOp); #if TENSORFLOW_USE_SYCL #define REGISTER(type) \ |