Merge pull request #14 from sparklxb/master

golang client

Author: tobegit3hub

golang_predict_client/src/generate_golang_files/model.proto

@@ -0,0 +1,22 @@
+syntax = "proto3";
+
+package tensorflow.serving;
+option cc_enable_arenas = true;
+
+import "google/protobuf/wrappers.proto";
+
+// Metadata for an inference request such as the model name and version.
+message ModelSpec {
+  // Required servable name.
+  string name = 1;
+
+  // Optional version. If unspecified, will use the latest (numerical) version.
+  // Typically not needed unless coordinating across multiple models that were
+  // co-trained and/or have inter-dependencies on the versions used at inference
+  // time.
+  google.protobuf.Int64Value version = 2;
+
+  // A named signature to evaluate. If unspecified, the default signature will
+  // be used. Note that only MultiInference will initially support this.
+  string signature_name = 3;
+}

golang_predict_client/src/generate_golang_files/predict.proto

@@ -0,0 +1,38 @@
+syntax = "proto3";
+
+package tensorflow.serving;
+option cc_enable_arenas = true;
+
+import "tensorflow/core/framework/tensor.proto";
+import "model.proto";
+
+// PredictRequest specifies which TensorFlow model to run, as well as
+// how inputs are mapped to tensors and how outputs are filtered before
+// returning to user.
+message PredictRequest {
+  // Model Specification.
+  ModelSpec model_spec = 1;
+
+  // Input tensors.
+  // Names of input tensor are alias names. The mapping from aliases to real
+  // input tensor names is expected to be stored as named generic signature
+  // under the key "inputs" in the model export.
+  // Each alias listed in a generic signature named "inputs" should be provided
+  // exactly once in order to run the prediction.
+  map<string, TensorProto> inputs = 2;
+
+  // Output filter.
+  // Names specified are alias names. The mapping from aliases to real output
+  // tensor names is expected to be stored as named generic signature under
+  // the key "outputs" in the model export.
+  // Only tensors specified here will be run/fetched and returned, with the
+  // exception that when none is specified, all tensors specified in the
+  // named signature will be run/fetched and returned.
+  repeated string output_filter = 3;
+}
+
+// Response for PredictRequest on successful run.
+message PredictResponse {
+  // Output tensors.
+  map<string, TensorProto> outputs = 1;
+}

golang_predict_client/src/generate_golang_files/prediction_service.proto

@@ -0,0 +1,13 @@
+syntax = "proto3";
+
+package tensorflow.serving;
+option cc_enable_arenas = true;
+
+import "predict.proto";
+
+// PredictionService provides access to machine-learned models loaded by
+// model_servers.
+service PredictionService {
+  // Predict -- provides access to loaded TensorFlow model.
+  rpc Predict(PredictRequest) returns (PredictResponse);
+}

golang_predict_client/src/generate_golang_files/tensorflow/contrib/ios_examples/.gitignore

@@ -0,0 +1,4 @@
+project.xcworkspace
+xcuserdata
+imagenet_comp_graph_label_strings.txt
+tensorflow_inception_graph.pb

golang_predict_client/src/generate_golang_files/tensorflow/contrib/makefile/.gitignore

@@ -0,0 +1,2 @@
+gen/
+downloads/

golang_predict_client/src/generate_golang_files/tensorflow/contrib/pi_examples/.gitignore

@@ -0,0 +1,4 @@
+tensorflow_inception_graph.pb
+imagenet_comp_graph_label_strings.txt
+tensorflow_inception_stripped.pb
+*/gen/

golang_predict_client/src/generate_golang_files/tensorflow/contrib/session_bundle/manifest.proto

@@ -0,0 +1,70 @@
+syntax = "proto3";
+
+package tensorflow.serving;
+
+// Signatures of model export.
+message Signatures {
+  // Default signature of the graph.
+  // WARNING(break-tutorial-inline-code): The following code snippet is
+  // in-lined in tutorials, please update tutorial documents accordingly
+  // whenever code changes.
+  Signature default_signature = 1;
+
+  // Named signatures of the graph.
+  map<string, Signature> named_signatures = 2;
+};
+
+// A binding to a tensor including the name and, possibly in the future, type
+// or other metadata. For example, this may specify whether a tensor supports
+// batch vs single inference.
+message TensorBinding {
+  // The name of the tensor to bind to.
+  string tensor_name = 1;
+};
+
+// An asset file or set of sharded files with the same name that will be bound
+// to a tensor at init / session_bundle load time.
+message AssetFile {
+  // The tensor to bind the asset filename to.
+  TensorBinding tensor_binding = 1;
+  // The filename within the assets directory. Note: does not include the base
+  // path or asset directory prefix. Base paths can and will change when models
+  // are deployed for serving.
+  string filename = 2;
+}
+
+// A Signature specifies the inputs and outputs of commonly used graphs.
+message Signature {
+  oneof type {
+    RegressionSignature regression_signature = 1;
+    ClassificationSignature classification_signature = 2;
+    GenericSignature generic_signature = 3;
+  }
+};
+
+// RegressionSignature specifies a graph that takes an input and returns an
+// output.
+message RegressionSignature {
+  TensorBinding input = 1;
+  TensorBinding output = 2;
+};
+
+// ClassificationSignature specifies a graph that takes an input and returns
+// classes and their scores.
+// WARNING(break-tutorial-inline-code): The following code snippet is
+// in-lined in tutorials, please update tutorial documents accordingly
+// whenever code changes.
+message ClassificationSignature {
+  TensorBinding input = 1;
+  TensorBinding classes = 2;
+  TensorBinding scores = 3;
+};
+
+// GenericSignature specifies a map from logical name to Tensor name.
+// Typical application of GenericSignature is to use a single GenericSignature
+// that includes all of the Tensor nodes and target names that may be useful at
+// serving, analysis or debugging time. The recommended name for this signature
+// in the ModelManifest is "generic_bindings".
+message GenericSignature {
+  map<string, TensorBinding> map = 1;
+};

golang_predict_client/src/generate_golang_files/tensorflow/contrib/tensorboard/plugins/projector/projector_config.proto

@@ -0,0 +1,46 @@
+/* Copyright 2016 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+syntax = "proto3";
+
+package tensorflow;
+
+message SpriteMetadata {
+  string image_path = 1;
+  // [width, height] of a single image in the sprite.
+  repeated uint32 single_image_dim = 2;
+}
+
+message EmbeddingInfo {
+  string tensor_name = 1;
+  string metadata_path = 2;
+  string bookmarks_path = 3;
+  // Shape of the 2D tensor [N x D]. If missing, it will be inferred from the
+  // model checkpoint.
+  repeated uint32 tensor_shape = 4;
+  SpriteMetadata sprite = 5;
+  // Path to the TSV file holding the tensor values. If missing, the tensor
+  // is assumed to be stored in the model checkpoint.
+  string tensor_path = 6;
+}
+
+message ProjectorConfig {
+  // Path to the checkpoint file. Use either this or model_checkpoint_dir.
+  string model_checkpoint_path = 1;
+  repeated EmbeddingInfo embeddings = 2;
+  // Path to the checkpoint directory. The directory will be scanned for the
+  // latest checkpoint file.
+  string model_checkpoint_dir = 3;
+}

golang_predict_client/src/generate_golang_files/tensorflow/core/debug/debug_service.proto

@@ -0,0 +1,39 @@
+/* Copyright 2016 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+syntax = "proto3";
+
+package tensorflow;
+
+import "tensorflow/core/util/event.proto";
+
+// Reply message from EventListener to the client, i.e., to the source of the
+// Event protocal buffers, e.g., debug ops inserted by a debugged runtime to a
+// TensorFlow graph being executed.
+message EventReply {
+}
+
+// EventListener: Receives Event protos, e.g., from debugged TensorFlow
+// runtime(s).
+service EventListener {
+  // Client(s) can use this RPC method to send the EventListener Event protos.
+  // The Event protos can hold information such as:
+  //   1) intermediate tensors from a debugged graph being executed, which can
+  //      be sent from DebugIdentity ops configured with grpc URLs.
+  //   2) GraphDefs of partition graphs, which can be sent from special debug
+  //      ops that get executed immediately after the beginning of the graph
+  //      execution.
+  rpc SendEvents(stream Event) returns (stream EventReply);
+}