From 91e4a19c1bbded4430299c636486c86b3fff2ab9 Mon Sep 17 00:00:00 2001 From: Derek Murray Date: Wed, 19 Apr 2017 19:45:54 -0800 Subject: 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 --- tensorflow/core/kernels/function_ops.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'tensorflow/core/kernels/function_ops.cc') 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) \ -- cgit v1.2.3