diff options
author | 2016-12-13 13:55:48 -0800 | |
---|---|---|
committer | 2016-12-13 14:05:04 -0800 | |
commit | df5d72d5452190e826fb57bf9ae955591b9b73a5 (patch) | |
tree | 2800d17eb480b4a860ac69f9cf1abb26f7b404bd | |
parent | 2b771b9fc34190d0acea439f9a930a078e54c37a (diff) |
Update comment to match current kernel registration macro.
Change: 141935716
-rw-r--r-- | tensorflow/core/framework/register_types.h | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/core/framework/register_types.h b/tensorflow/core/framework/register_types.h index cdf0d197df..d79468afa1 100644 --- a/tensorflow/core/framework/register_types.h +++ b/tensorflow/core/framework/register_types.h @@ -43,9 +43,10 @@ limitations under the License. // This can be used to register a different template instantiation of // an OpKernel for different signatures, e.g.: /* - #define REGISTER_PARTITION(type) \ - REGISTER_TF_OP_KERNEL("partition", DEVICE_CPU, #type ", int32", \ - PartitionOp<type>); + #define REGISTER_PARTITION(type) \ + REGISTER_KERNEL_BUILDER( \ + Name("Partition").Device(DEVICE_CPU).TypeConstraint<type>("T"), \ + PartitionOp<type>); TF_CALL_ALL_TYPES(REGISTER_PARTITION) #undef REGISTER_PARTITION */ |