1 /* Copyright 2016 The TensorFlow Authors. All Rights Reserved. 2 3 Licensed under the Apache License, Version 2.0 (the "License"); 4 you may not use this file except in compliance with the License. 5 You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9 Unless required by applicable law or agreed to in writing, software 10 distributed under the License is distributed on an "AS IS" BASIS, 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 See the License for the specific language governing permissions and 13 limitations under the License. 14 ==============================================================================*/ 15 16 #ifndef TENSORFLOW_CC_SAVED_MODEL_SIGNATURE_CONSTANTS_H_ 17 #define TENSORFLOW_CC_SAVED_MODEL_SIGNATURE_CONSTANTS_H_ 18 19 namespace tensorflow { 20 21 /// Key in the signature def map for `default` serving signatures. The default 22 /// signature is used in inference requests where a specific signature was not 23 /// specified. 24 static constexpr char kDefaultServingSignatureDefKey[] = "serving_default"; 25 26 //////////////////////////////////////////////////////////////////////////////// 27 /// Classification API constants. 28 29 /// Classification inputs. 30 static constexpr char kClassifyInputs[] = "inputs"; 31 32 /// Classification method name used in a SignatureDef. 33 static constexpr char kClassifyMethodName[] = "tensorflow/serving/classify"; 34 35 /// Classification classes output. 36 static constexpr char kClassifyOutputClasses[] = "classes"; 37 38 /// Classification scores output. 39 static constexpr char kClassifyOutputScores[] = "scores"; 40 41 //////////////////////////////////////////////////////////////////////////////// 42 /// Predict API constants. 43 44 /// Predict inputs. 45 static constexpr char kPredictInputs[] = "inputs"; 46 47 /// Predict method name used in a SignatureDef. 48 static constexpr char kPredictMethodName[] = "tensorflow/serving/predict"; 49 50 /// Predict outputs. 51 static constexpr char kPredictOutputs[] = "outputs"; 52 53 //////////////////////////////////////////////////////////////////////////////// 54 /// Regression API constants. 55 56 /// Regression inputs. 57 static constexpr char kRegressInputs[] = "inputs"; 58 59 /// Regression method name used in a SignatureDef. 60 static constexpr char kRegressMethodName[] = "tensorflow/serving/regress"; 61 62 /// Regression outputs. 63 static constexpr char kRegressOutputs[] = "outputs"; 64 65 //////////////////////////////////////////////////////////////////////////////// 66 67 } // namespace tensorflow 68 69 #endif // TENSORFLOW_CC_SAVED_MODEL_SIGNATURE_CONSTANTS_H_ 70
