diff options
author | 2017-10-10 08:36:23 -0700 | |
---|---|---|
committer | 2017-10-10 08:39:56 -0700 | |
commit | 3bafe0a86f67dd54197c6d60bdb5053f510de7d8 (patch) | |
tree | b7e9b7fd922f1c5d96ebf04465cfb45154964b4c /tensorflow/python/__init__.py | |
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/python/__init__.py')
-rw-r--r-- | tensorflow/python/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/__init__.py b/tensorflow/python/__init__.py index f3bdea92dd..f21f1f822c 100644 --- a/tensorflow/python/__init__.py +++ b/tensorflow/python/__init__.py @@ -213,6 +213,8 @@ _allowed_symbols.extend([ 'quint16', 'quint8', 'string', + 'uint64', + 'uint32', 'uint16', 'uint8', 'resource', |