diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-11-21 12:24:12 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-21 12:27:46 -0800 |
commit | 6c7bd707ce26cc89d542bbb326882026a613748c (patch) | |
tree | 71672568b60e68f5aa5c1fa5ec5f1014fdd2d074 /tensorflow/cc/saved_model | |
parent | b5dcb0161942c467be6cba19aa0ee05aef742d2e (diff) |
Add tpu saved model tags. No cpu tag is added because cpu is assumed to be the implicit device.
PiperOrigin-RevId: 176544698
Diffstat (limited to 'tensorflow/cc/saved_model')
-rw-r--r-- | tensorflow/cc/saved_model/tag_constants.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/cc/saved_model/tag_constants.h b/tensorflow/cc/saved_model/tag_constants.h index 2b0b2d5c7f..b71cb263ca 100644 --- a/tensorflow/cc/saved_model/tag_constants.h +++ b/tensorflow/cc/saved_model/tag_constants.h @@ -21,6 +21,9 @@ namespace tensorflow { /// Tag for the `gpu` graph. constexpr char kSavedModelTagGpu[] = "gpu"; +/// Tag for the `tpu` graph. +constexpr char kSavedModelTagTpu[] = "tpu"; + /// Tag for the `serving` graph. constexpr char kSavedModelTagServe[] = "serve"; |