diff options
| author |  Sukriti Ramesh <sukritiramesh@google.com> | 2017-06-20 12:59:58 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-20 13:03:57 -0700 |
| commit | ad2faeac06c7be527108d83cd8771b6c0f296257 (patch) | |
| tree | e1d989d1a9bb085fb49b9697e47b8555ddfd6b17 /tensorflow/cc/saved_model | |
| parent | 716687c091aebb4cace35229fb150c356f0079c8 (diff) | |
Add SavedModel tag-constant for gpu.
PiperOrigin-RevId: 159600621
Diffstat (limited to 'tensorflow/cc/saved_model')
| -rw-r--r-- | tensorflow/cc/saved_model/tag_constants.h | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/cc/saved_model/tag_constants.h b/tensorflow/cc/saved_model/tag_constants.h index 48ab1158e4..2b0b2d5c7f 100644 --- a/tensorflow/cc/saved_model/tag_constants.h +++ b/tensorflow/cc/saved_model/tag_constants.h @@ -18,10 +18,13 @@ limitations under the License. namespace tensorflow { +/// Tag for the `gpu` graph. +constexpr char kSavedModelTagGpu[] = "gpu"; + /// Tag for the `serving` graph. constexpr char kSavedModelTagServe[] = "serve"; -/// Tag for the `training` graph.` +/// Tag for the `training` graph. constexpr char kSavedModelTagTrain[] = "train"; } // namespace tensorflow |