diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-05-31 08:37:09 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-05-31 09:47:15 -0700 |
| commit | 2c36226b6671367f681746f8cf850b91d2c17883 (patch) | |
| tree | 342ad60f5007df86b539db116369b31db0da7c54 /tensorflow/core/lib/strings/strcat.h | |
| parent | bfbd4806323123706c902e9a3cd0b3ff061db371 (diff) | |
Made the summary operations work with fp16
Change: 123649866
Diffstat (limited to 'tensorflow/core/lib/strings/strcat.h')
| -rw-r--r-- | tensorflow/core/lib/strings/strcat.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/lib/strings/strcat.h b/tensorflow/core/lib/strings/strcat.h index d7d5352a88..bbf33742c7 100644 --- a/tensorflow/core/lib/strings/strcat.h +++ b/tensorflow/core/lib/strings/strcat.h @@ -27,6 +27,10 @@ limitations under the License. #include "tensorflow/core/platform/macros.h" #include "tensorflow/core/platform/types.h" +namespace Eigen { +struct half; +} + // The AlphaNum type was designed to be used as the parameter type for StrCat(). // Any routine accepting either a string or a number may accept it. // The basic idea is that by accepting a "const AlphaNum &" as an argument @@ -118,6 +122,7 @@ class AlphaNum { AlphaNum(double f) // NOLINT(runtime/explicit) : piece_(digits_, strlen(DoubleToBuffer(f, digits_))) {} + AlphaNum(const Eigen::half &f); // NOLINT(runtime/explicit) AlphaNum(Hex hex); // NOLINT(runtime/explicit) AlphaNum(const char *c_str) : piece_(c_str) {} // NOLINT(runtime/explicit) |