diff options
author | Peter Hawkins <phawkins@google.com> | 2017-12-04 15:13:45 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-04 15:19:56 -0800 |
commit | d87a76d7d6b053219dbd49a87b3c4b379a1c6566 (patch) | |
tree | baba15d97d64bd3f54cc06644b611b00788dc931 /tensorflow/python/tensorflow.i | |
parent | f5f7b85b3c10da291f23263f0b4f0ac7cffa9059 (diff) |
Enable bfloat16 use from Python:
* add a bfloat16 Python type and NumPy extension.
* allow the bfloat16 type in a number places in the Python libraries.
PiperOrigin-RevId: 177875784
Diffstat (limited to 'tensorflow/python/tensorflow.i')
-rw-r--r-- | tensorflow/python/tensorflow.i | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/tensorflow.i b/tensorflow/python/tensorflow.i index d221dd523b..344702097f 100644 --- a/tensorflow/python/tensorflow.i +++ b/tensorflow/python/tensorflow.i @@ -33,6 +33,8 @@ limitations under the License. %include "tensorflow/python/client/tf_session.i" %include "tensorflow/python/client/device_lib.i" +%include "tensorflow/python/lib/core/bfloat16.i" + %include "tensorflow/python/lib/io/file_io.i" %include "tensorflow/python/training/quantize_training.i" %include "tensorflow/python/training/server_lib.i" |