diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-19 15:40:18 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-19 15:45:12 -0700 |
| commit | c96841dbd199d3c1a15a89e8c44c7c1d164968b9 (patch) | |
| tree | 392511a31e3565f862c7257da6a35c31b6587968 /tensorflow/core/BUILD | |
| parent | b2b98a5ad1b647b77cb42761671cd9b3cf0e87b6 (diff) | |
This CL adds a new `tf.print` operator that more closely aligns with the standard python `print` method, and deprecates the old `tf.Print` operator (to be removed in in v2.0).
It follows the design doc specified in https://github.com/tensorflow/community/pull/14 and additionally incorporates the community feedback and design review decisions.
This CL adds two new internal graph operators: a StringFormat operator that formats a template string with a list of input tensors to insert into the string and outputs a string scalar containing the result, and a PrintV2 operator that prints a string scalar to a specified output stream or logging level.
The formatting op is exposed at `tf.strings.Format`. A new python method is exposed at `tf.print` that takes a list of inputs that may be nested structures and may contain tensors, formats them nicely using the formatting op, and returns a PrintV2 operator that prints them. In Eager mode and inside defuns this PrintV2 operator will automatically be executed, but in graph mode it will need to be either added to `sess.run`, or used as a control dependency for other operators being executed.
As compared to the previous print function, the new print function:
- Has an API that more closely aligns with the standard python3 print
- Supports changing the print logging level/output stream
- allows printing arbitrary (optionally nested) data structures as opposed to just flat lists of tensors
- support printing sparse tensors
- changes printed tensor format to show more meaningful summary (recursively print the first and last elements of each tensor dimension, instead of just the first few elements of the tensor irregardless of dimension).
PiperOrigin-RevId: 213709924
Diffstat (limited to 'tensorflow/core/BUILD')
| -rw-r--r-- | tensorflow/core/BUILD | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 4b2589aaeb..e82dd13b31 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -1067,7 +1067,6 @@ tf_gen_op_libs( "spectral_ops", "state_ops", "stateless_random_ops", - "string_ops", "summary_ops", "training_ops", ], @@ -1075,6 +1074,13 @@ tf_gen_op_libs( tf_gen_op_libs( op_lib_names = [ + "string_ops", + ], + deps = ["@com_google_absl//absl/strings"], +) + +tf_gen_op_libs( + op_lib_names = [ "array_ops", ], deps = [":protos_all_cc"], |