diff options
| author |  Peter Hawkins <phawkins@google.com> | 2017-10-10 08:36:23 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-10 08:39:56 -0700 |
| commit | 3bafe0a86f67dd54197c6d60bdb5053f510de7d8 (patch) | |
| tree | b7e9b7fd922f1c5d96ebf04465cfb45154964b4c /tensorflow/c/c_api.h | |
| parent | 5a26d1ede506825455d1199267d88caeba7d206a (diff) | |
Add uint32 and uint64 types to TensorFlow.
This change merely creates the types, but does not register kernels that act on uint32/uint64 values. It also does not alter most op registration lists to include uint32/uint64 values. If desirable, that can be done in a subsequent change, although binary size will likely prove problematic if adding more kernels.
The intent of the change is so XLA-compiled code can make use uint32/uint64 types. Since XLA does not use traditional TensorFlow kernels, using uint32/uint64 operators from XLA will require only uint32/uint64 op registrations, but will require few new kernel registrations.
PiperOrigin-RevId: 171681867
Diffstat (limited to 'tensorflow/c/c_api.h')
| -rw-r--r-- | tensorflow/c/c_api.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/c/c_api.h b/tensorflow/c/c_api.h index db94828e1a..68a758498d 100644 --- a/tensorflow/c/c_api.h +++ b/tensorflow/c/c_api.h @@ -118,6 +118,8 @@ typedef enum TF_DataType { TF_HALF = 19, TF_RESOURCE = 20, TF_VARIANT = 21, + TF_UINT32 = 22, + TF_UINT64 = 23, } TF_DataType; // TF_DataTypeSize returns the sizeof() for the underlying type corresponding |