int64 step_id = 1;
12 | */
13 | long getStepId();
14 | }
15 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/util/status.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | option cc_enable_arenas = true;
4 |
5 | import "tensorflow/core/lib/core/error_codes.proto";
6 |
7 | package tensorflow.serving;
8 |
9 | // Status that corresponds to Status in
10 | // third_party/tensorflow/core/lib/core/status.h.
11 | message StatusProto {
12 | // Error code.
13 | error.Code error_code = 1 [json_name = "error_code"];
14 |
15 | // Error message. Will only be set if an error was encountered.
16 | string error_message = 2 [json_name = "error_message"];
17 | }
18 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/servables/tensorflow/saved_model_bundle_source_adapter.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | import "tensorflow_serving/servables/tensorflow/session_bundle_config.proto";
4 |
5 | package tensorflow.serving;
6 |
7 | // Config proto for SavedModelBundleSourceAdapter.
8 | message SavedModelBundleSourceAdapterConfig {
9 | // A SessionBundleConfig.
10 | // FOR INTERNAL USE ONLY DURING TRANSITION TO SAVED_MODEL. WILL BE DEPRECATED.
11 | // TODO(b/32248363): Replace this field with the "real" field(s).
12 | SessionBundleConfig legacy_config = 1000;
13 | }
14 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/config/logging_config.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow.serving;
4 | option cc_enable_arenas = true;
5 |
6 | import "tensorflow_serving/config/log_collector_config.proto";
7 |
8 | message SamplingConfig {
9 | // Requests will be logged uniformly at random with this probability. Valid
10 | // range: [0, 1.0].
11 | double sampling_rate = 1;
12 | }
13 |
14 | // Configuration for logging query/responses.
15 | message LoggingConfig {
16 | LogCollectorConfig log_collector_config = 1;
17 | SamplingConfig sampling_config = 2;
18 | }
19 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/config/monitoring_config.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow.serving;
4 | option cc_enable_arenas = true;
5 |
6 | // Configuration for Prometheus monitoring.
7 | message PrometheusConfig {
8 | // Whether to expose Prometheus metrics.
9 | bool enable = 1;
10 |
11 | // The endpoint to expose Prometheus metrics.
12 | // If not specified, PrometheusExporter::kPrometheusPath value is used.
13 | string path = 2;
14 | }
15 |
16 | // Configuration for monitoring.
17 | message MonitoringConfig {
18 | PrometheusConfig prometheus_config = 1;
19 | }
20 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/MakeCallableResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface MakeCallableResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.MakeCallableResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * A handle to the created callable. 13 | *14 | * 15 | *
int64 handle = 1;
16 | */
17 | long getHandle();
18 | }
19 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/StepSequenceOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface StepSequenceOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.StepSequence)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int64 graph_key = 1;
12 | */
13 | long getGraphKey();
14 |
15 | /**
16 | * int64 next_step_id = 2;
17 | */
18 | long getNextStepId();
19 | }
20 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/reader_base.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "ReaderBaseProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 |
10 | // For serializing and restoring the state of ReaderBase, see
11 | // reader_base.h for details.
12 | message ReaderBaseState {
13 | int64 work_started = 1;
14 | int64 work_finished = 2;
15 | int64 num_records_produced = 3;
16 | bytes current_work = 4;
17 | };
18 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/AutoParallelOptionsOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/rewriter_config.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface AutoParallelOptionsOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.AutoParallelOptions)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * bool enable = 1;
12 | */
13 | boolean getEnable();
14 |
15 | /**
16 | * int32 num_replicas = 2;
17 | */
18 | int getNumReplicas();
19 | }
20 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/GraphTransferNodeInputOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/graph_transfer_info.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface GraphTransferNodeInputOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GraphTransferNodeInput)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int32 node_id = 1;
12 | */
13 | int getNodeId();
14 |
15 | /**
16 | * int32 output_port = 2;
17 | */
18 | int getOutputPort();
19 | }
20 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/CompleteInstanceResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface CompleteInstanceResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CompleteInstanceResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int32 instance_key = 1;
12 | */
13 | int getInstanceKey();
14 |
15 | /**
16 | * int32 source_rank = 2;
17 | */
18 | int getSourceRank();
19 | }
20 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/iterator.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "IteratorProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.util";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 |
10 | // Protocol buffer representing the metadata for an iterator's state stored
11 | // as a Variant tensor.
12 | message IteratorStateMetadata {
13 | // A user-specified version string.
14 | string version = 1;
15 |
16 | // Keys for tensors in the VariantTensorDataProto.
17 | repeated string keys = 2;
18 | }
19 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/config/platform_config.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow.serving;
4 | option cc_enable_arenas = true;
5 |
6 | import "google/protobuf/any.proto";
7 |
8 | // Configuration for a servable platform e.g. tensorflow or other ML systems.
9 | message PlatformConfig {
10 | // The config proto for a SourceAdapter in the StoragePathSourceAdapter
11 | // registry.
12 | google.protobuf.Any source_adapter_config = 1;
13 | };
14 |
15 | message PlatformConfigMap {
16 | // A map from a platform name to a platform config. The platform name is used
17 | // in ModelConfig.model_platform.
18 | map12 | * The new version number for the extended graph, to be used in the next call 13 | * to ExtendSession. 14 | *15 | * 16 | *
int64 new_graph_version = 4;
17 | */
18 | long getNewGraphVersion();
19 | }
20 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/BytesListOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/feature.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface BytesListOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.BytesList)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated bytes value = 1;
12 | */
13 | java.util.Listrepeated bytes value = 1;
16 | */
17 | int getValueCount();
18 | /**
19 | * repeated bytes value = 1;
20 | */
21 | com.google.protobuf.ByteString getValue(int index);
22 | }
23 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/Int64ListOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/feature.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface Int64ListOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.Int64List)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated int64 value = 1 [packed = true];
12 | */
13 | java.util.Listrepeated int64 value = 1 [packed = true];
16 | */
17 | int getValueCount();
18 | /**
19 | * repeated int64 value = 1 [packed = true];
20 | */
21 | long getValue(int index);
22 | }
23 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/FloatListOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/feature.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface FloatListOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.FloatList)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated float value = 1 [packed = true];
12 | */
13 | java.util.Listrepeated float value = 1 [packed = true];
16 | */
17 | int getValueCount();
18 | /**
19 | * repeated float value = 1 [packed = true];
20 | */
21 | float getValue(int index);
22 | }
23 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/ExampleOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/example.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface ExampleOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.Example)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * .tensorflow.Features features = 1;
12 | */
13 | boolean hasFeatures();
14 | /**
15 | * .tensorflow.Features features = 1;
16 | */
17 | org.tensorflow.example.Features getFeatures();
18 | /**
19 | * .tensorflow.Features features = 1;
20 | */
21 | org.tensorflow.example.FeaturesOrBuilder getFeaturesOrBuilder();
22 | }
23 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/GetStepSequenceRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface GetStepSequenceRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GetStepSequenceRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated int64 graph_key = 1;
12 | */
13 | java.util.Listrepeated int64 graph_key = 1;
16 | */
17 | int getGraphKeyCount();
18 | /**
19 | * repeated int64 graph_key = 1;
20 | */
21 | long getGraphKey(int index);
22 | }
23 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/TracingRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface TracingRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.TracingRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * .tensorflow.TraceOpts options = 1;
12 | */
13 | boolean hasOptions();
14 | /**
15 | * .tensorflow.TraceOpts options = 1;
16 | */
17 | org.tensorflow.distruntime.TraceOpts getOptions();
18 | /**
19 | * .tensorflow.TraceOpts options = 1;
20 | */
21 | org.tensorflow.distruntime.TraceOptsOrBuilder getOptionsOrBuilder();
22 | }
23 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/AllocationRecordOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/step_stats.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface AllocationRecordOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.AllocationRecord)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The timestamp of the operation. 13 | *14 | * 15 | *
int64 alloc_micros = 1;
16 | */
17 | long getAllocMicros();
18 |
19 | /**
20 | * 21 | * Number of bytes allocated, or de-allocated if negative. 22 | *23 | * 24 | *
int64 alloc_bytes = 2;
25 | */
26 | long getAllocBytes();
27 | }
28 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/InterconnectLinkOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/device_attributes.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface InterconnectLinkOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.InterconnectLink)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int32 device_id = 1;
12 | */
13 | int getDeviceId();
14 |
15 | /**
16 | * string type = 2;
17 | */
18 | java.lang.String getType();
19 | /**
20 | * string type = 2;
21 | */
22 | com.google.protobuf.ByteString
23 | getTypeBytes();
24 |
25 | /**
26 | * int32 strength = 3;
27 | */
28 | int getStrength();
29 | }
30 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/tensor_description.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "TensorDescriptionProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 | import "tensorflow/core/framework/types.proto";
10 | import "tensorflow/core/framework/tensor_shape.proto";
11 | import "tensorflow/core/framework/allocation_description.proto";
12 |
13 | message TensorDescription {
14 | // Data type of tensor elements
15 | DataType dtype = 1;
16 |
17 | // Shape of the tensor.
18 | TensorShapeProto shape = 2;
19 |
20 | // Information about the size and allocator used for the data
21 | AllocationDescription allocation_description = 4;
22 | };
23 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/ReaderBaseStateOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/reader_base.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface ReaderBaseStateOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ReaderBaseState)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int64 work_started = 1;
12 | */
13 | long getWorkStarted();
14 |
15 | /**
16 | * int64 work_finished = 2;
17 | */
18 | long getWorkFinished();
19 |
20 | /**
21 | * int64 num_records_produced = 3;
22 | */
23 | long getNumRecordsProduced();
24 |
25 | /**
26 | * bytes current_work = 4;
27 | */
28 | com.google.protobuf.ByteString getCurrentWork();
29 | }
30 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/ExecutorOptsOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface ExecutorOptsOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ExecutorOpts)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * bool record_costs = 1;
12 | */
13 | boolean getRecordCosts();
14 |
15 | /**
16 | * bool record_timeline = 3;
17 | */
18 | boolean getRecordTimeline();
19 |
20 | /**
21 | * bool record_partition_graphs = 4;
22 | */
23 | boolean getRecordPartitionGraphs();
24 |
25 | /**
26 | * bool report_tensor_allocations_upon_oom = 5;
27 | */
28 | boolean getReportTensorAllocationsUponOom();
29 | }
30 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/CriticalSectionDefOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/critical_section.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface CriticalSectionDefOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CriticalSectionDef)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Name of the critical section handle. 13 | *14 | * 15 | *
string critical_section_name = 1;
16 | */
17 | java.lang.String getCriticalSectionName();
18 | /**
19 | * 20 | * Name of the critical section handle. 21 | *22 | * 23 | *
string critical_section_name = 1;
24 | */
25 | com.google.protobuf.ByteString
26 | getCriticalSectionNameBytes();
27 | }
28 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/DeleteWorkerSessionRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface DeleteWorkerSessionRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.DeleteWorkerSessionRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Sessions are identified by a given handle. 13 | *14 | * 15 | *
string session_handle = 1;
16 | */
17 | java.lang.String getSessionHandle();
18 | /**
19 | * 20 | * Sessions are identified by a given handle. 21 | *22 | * 23 | *
string session_handle = 1;
24 | */
25 | com.google.protobuf.ByteString
26 | getSessionHandleBytes();
27 | }
28 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/allocation_description.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "AllocationDescriptionProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 |
10 | message AllocationDescription {
11 | // Total number of bytes requested
12 | int64 requested_bytes = 1;
13 |
14 | // Total number of bytes allocated if known
15 | int64 allocated_bytes = 2;
16 |
17 | // Name of the allocator used
18 | string allocator_name = 3;
19 |
20 | // Identifier of the allocated buffer if known
21 | int64 allocation_id = 4;
22 |
23 | // Set if this tensor only has one remaining reference
24 | bool has_single_reference = 5;
25 |
26 | // Address of the allocation.
27 | uint64 ptr = 6;
28 | };
29 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/protobuf/saved_model.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "SavedModelProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
9 | import "tensorflow/core/protobuf/meta_graph.proto";
10 |
11 | // SavedModel is the high level serialization format for TensorFlow Models.
12 | // See [todo: doc links, similar to session_bundle] for more information.
13 | message SavedModel {
14 | // The schema version of the SavedModel instance. Used for versioning when
15 | // making future changes to the specification/implementation. Initial value
16 | // at release will be 1.
17 | int64 saved_model_schema_version = 1;
18 |
19 | // One or more MetaGraphs.
20 | repeated MetaGraphDef meta_graphs = 2;
21 | }
22 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/protobuf/named_tensor.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "NamedTensorProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
9 | import "tensorflow/core/framework/tensor.proto";
10 |
11 | // A pair of tensor name and tensor values.
12 | message NamedTensorProto {
13 | // Name of the tensor.
14 | string name = 1;
15 |
16 | // The client can populate a TensorProto using a tensorflow::Tensor`, or
17 | // directly using the protobuf field accessors.
18 | //
19 | // The client specifies whether the returned tensor values should be
20 | // filled tensor fields (float_val, int_val, etc.) or encoded in a
21 | // compact form in tensor.tensor_content.
22 | TensorProto tensor = 2;
23 | }
24 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/LabeledStepStatsOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface LabeledStepStatsOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.LabeledStepStats)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int64 step_id = 1;
12 | */
13 | long getStepId();
14 |
15 | /**
16 | * .tensorflow.StepStats step_stats = 2;
17 | */
18 | boolean hasStepStats();
19 | /**
20 | * .tensorflow.StepStats step_stats = 2;
21 | */
22 | org.tensorflow.framework.StepStats getStepStats();
23 | /**
24 | * .tensorflow.StepStats step_stats = 2;
25 | */
26 | org.tensorflow.framework.StepStatsOrBuilder getStepStatsOrBuilder();
27 | }
28 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/GraphTransferNodeOutputInfoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/graph_transfer_info.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface GraphTransferNodeOutputInfoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GraphTransferNodeOutputInfo)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int32 node_id = 1;
12 | */
13 | int getNodeId();
14 |
15 | /**
16 | * repeated int32 max_byte_size = 2;
17 | */
18 | java.util.Listrepeated int32 max_byte_size = 2;
21 | */
22 | int getMaxByteSizeCount();
23 | /**
24 | * repeated int32 max_byte_size = 2;
25 | */
26 | int getMaxByteSize(int index);
27 | }
28 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/protobuf/critical_section.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "CriticalSectionProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
9 |
10 | // Protocol buffer representing a CriticalSection.
11 | message CriticalSectionDef {
12 | // Name of the critical section handle.
13 | string critical_section_name = 1;
14 | }
15 |
16 | // Protocol buffer representing a CriticalSection execution.
17 | message CriticalSectionExecutionDef {
18 | // Name of the critical section handle.
19 | string execute_in_critical_section_name = 1;
20 | // Whether this operation requires exclusive access to its resources,
21 | // (i.e., no other CriticalSections may request the same resources).
22 | bool exclusive_resource_access = 2;
23 | }
24 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/NodeOutputOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/step_stats.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface NodeOutputOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.NodeOutput)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int32 slot = 1;
12 | */
13 | int getSlot();
14 |
15 | /**
16 | * .tensorflow.TensorDescription tensor_description = 3;
17 | */
18 | boolean hasTensorDescription();
19 | /**
20 | * .tensorflow.TensorDescription tensor_description = 3;
21 | */
22 | org.tensorflow.framework.TensorDescription getTensorDescription();
23 | /**
24 | * .tensorflow.TensorDescription tensor_description = 3;
25 | */
26 | org.tensorflow.framework.TensorDescriptionOrBuilder getTensorDescriptionOrBuilder();
27 | }
28 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/CloseSessionRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface CloseSessionRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CloseSessionRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * REQUIRED: session_handle must be returned by a CreateSession call 13 | * to the same master service. 14 | *15 | * 16 | *
string session_handle = 1;
17 | */
18 | java.lang.String getSessionHandle();
19 | /**
20 | * 21 | * REQUIRED: session_handle must be returned by a CreateSession call 22 | * to the same master service. 23 | *24 | * 25 | *
string session_handle = 1;
26 | */
27 | com.google.protobuf.ByteString
28 | getSessionHandleBytes();
29 | }
30 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/protobuf/queue_runner.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "QueueRunnerProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
9 | import "tensorflow/core/lib/core/error_codes.proto";
10 |
11 | // Protocol buffer representing a QueueRunner.
12 | message QueueRunnerDef {
13 | // Queue name.
14 | string queue_name = 1;
15 |
16 | // A list of enqueue operations.
17 | repeated string enqueue_op_name = 2;
18 |
19 | // The operation to run to close the queue.
20 | string close_op_name = 3;
21 |
22 | // The operation to run to cancel the queue.
23 | string cancel_op_name = 4;
24 |
25 | // A list of exception types considered to signal a safely closed queue
26 | // if raised during enqueue operations.
27 | repeated error.Code queue_closed_exception_types = 5;
28 | }
29 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/regression.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | option cc_enable_arenas = true;
4 |
5 | import "tensorflow_serving/apis/input.proto";
6 | import "tensorflow_serving/apis/model.proto";
7 |
8 | package tensorflow.serving;
9 |
10 | // Regression result for a single item (tensorflow.Example).
11 | message Regression {
12 | float value = 1;
13 | }
14 |
15 | // Contains one result per input example, in the same order as the input in
16 | // RegressionRequest.
17 | message RegressionResult {
18 | repeated Regression regressions = 1;
19 | }
20 |
21 | // RPC interfaces.
22 |
23 | message RegressionRequest {
24 | // Model Specification. If version is not specified, will use the latest
25 | // (numerical) version.
26 | ModelSpec model_spec = 1;
27 |
28 | // Input data.
29 | tensorflow.serving.Input input = 2;
30 | }
31 |
32 | message RegressionResponse {
33 | // Effective Model Specification used for regression.
34 | ModelSpec model_spec = 2;
35 |
36 | RegressionResult result = 1;
37 | }
38 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/MemoryLogStepOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/log_memory.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface MemoryLogStepOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.MemoryLogStep)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Process-unique step id. 13 | *14 | * 15 | *
int64 step_id = 1;
16 | */
17 | long getStepId();
18 |
19 | /**
20 | * 21 | * Handle describing the feeds and fetches of the step. 22 | *23 | * 24 | *
string handle = 2;
25 | */
26 | java.lang.String getHandle();
27 | /**
28 | * 29 | * Handle describing the feeds and fetches of the step. 30 | *31 | * 32 | *
string handle = 2;
33 | */
34 | com.google.protobuf.ByteString
35 | getHandleBytes();
36 | }
37 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/SummaryDescriptionOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/summary.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface SummaryDescriptionOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.SummaryDescription)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Hint on how plugins should process the data in this series. 13 | * Supported values include "scalar", "histogram", "image", "audio" 14 | *15 | * 16 | *
string type_hint = 1;
17 | */
18 | java.lang.String getTypeHint();
19 | /**
20 | * 21 | * Hint on how plugins should process the data in this series. 22 | * Supported values include "scalar", "histogram", "image", "audio" 23 | *24 | * 25 | *
string type_hint = 1;
26 | */
27 | com.google.protobuf.ByteString
28 | getTypeHintBytes();
29 | }
30 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/model_service.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | option cc_enable_arenas = true;
4 |
5 | import "tensorflow_serving/apis/get_model_status.proto";
6 | import "tensorflow_serving/apis/model_management.proto";
7 |
8 | package tensorflow.serving;
9 |
10 | // ModelService provides methods to query and update the state of the server,
11 | // e.g. which models/versions are being served.
12 | service ModelService {
13 | // Gets status of model. If the ModelSpec in the request does not specify
14 | // version, information about all versions of the model will be returned. If
15 | // the ModelSpec in the request does specify a version, the status of only
16 | // that version will be returned.
17 | rpc GetModelStatus(GetModelStatusRequest) returns (GetModelStatusResponse);
18 |
19 | // Reloads the set of served models. The new config supersedes the old one,
20 | // so if a model is omitted from the new config it will be unloaded and no
21 | // longer served.
22 | rpc HandleReloadConfigRequest(ReloadConfigRequest)
23 | returns (ReloadConfigResponse);
24 | }
25 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/OpDeprecationOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/op_def.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface OpDeprecationOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.OpDeprecation)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * First GraphDef version at which the op is disallowed. 13 | *14 | * 15 | *
int32 version = 1;
16 | */
17 | int getVersion();
18 |
19 | /**
20 | * 21 | * Explanation of why it was deprecated and what to use instead. 22 | *23 | * 24 | *
string explanation = 2;
25 | */
26 | java.lang.String getExplanation();
27 | /**
28 | * 29 | * Explanation of why it was deprecated and what to use instead. 30 | *31 | * 32 | *
string explanation = 2;
33 | */
34 | com.google.protobuf.ByteString
35 | getExplanationBytes();
36 | }
37 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/resource_handle.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "ResourceHandle";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 |
10 | // Protocol buffer representing a handle to a tensorflow resource. Handles are
11 | // not valid across executions, but can be serialized back and forth from within
12 | // a single run.
13 | message ResourceHandleProto {
14 | // Unique name for the device containing the resource.
15 | string device = 1;
16 |
17 | // Container in which this resource is placed.
18 | string container = 2;
19 |
20 | // Unique name of this resource.
21 | string name = 3;
22 |
23 | // Hash code for the type of the resource. Is only valid in the same device
24 | // and in the same execution.
25 | uint64 hash_code = 4;
26 |
27 | // For debug-only, the name of the type pointed to by this handle, if
28 | // available.
29 | string maybe_type_name = 5;
30 | };
31 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/RegisterGraphResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface RegisterGraphResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.RegisterGraphResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * If the registration succeeds, returns an opaque graph_handle to 13 | * the master. The master calls RunGraph with graph_handle to 14 | * compute different steps. 15 | *16 | * 17 | *
string graph_handle = 1;
18 | */
19 | java.lang.String getGraphHandle();
20 | /**
21 | * 22 | * If the registration succeeds, returns an opaque graph_handle to 23 | * the master. The master calls RunGraph with graph_handle to 24 | * compute different steps. 25 | *26 | * 27 | *
string graph_handle = 1;
28 | */
29 | com.google.protobuf.ByteString
30 | getGraphHandleBytes();
31 | }
32 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/MemoryLogTensorDeallocationOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/log_memory.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface MemoryLogTensorDeallocationOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.MemoryLogTensorDeallocation)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Id of the tensor buffer being deallocated, used to match to a 13 | * corresponding allocation. 14 | *15 | * 16 | *
int64 allocation_id = 1;
17 | */
18 | long getAllocationId();
19 |
20 | /**
21 | * 22 | * Name of the allocator used. 23 | *24 | * 25 | *
string allocator_name = 2;
26 | */
27 | java.lang.String getAllocatorName();
28 | /**
29 | * 30 | * Name of the allocator used. 31 | *32 | * 33 | *
string allocator_name = 2;
34 | */
35 | com.google.protobuf.ByteString
36 | getAllocatorNameBytes();
37 | }
38 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/OpListOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/op_def.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface OpListOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.OpList)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.OpDef op = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.OpDef op = 1;
17 | */
18 | org.tensorflow.framework.OpDef getOp(int index);
19 | /**
20 | * repeated .tensorflow.OpDef op = 1;
21 | */
22 | int getOpCount();
23 | /**
24 | * repeated .tensorflow.OpDef op = 1;
25 | */
26 | java.util.List
27 | getOpOrBuilderList();
28 | /**
29 | * repeated .tensorflow.OpDef op = 1;
30 | */
31 | org.tensorflow.framework.OpDefOrBuilder getOpOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/ApiDefsOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/api_def.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface ApiDefsOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ApiDefs)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.ApiDef op = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.ApiDef op = 1;
17 | */
18 | org.tensorflow.framework.ApiDef getOp(int index);
19 | /**
20 | * repeated .tensorflow.ApiDef op = 1;
21 | */
22 | int getOpCount();
23 | /**
24 | * repeated .tensorflow.ApiDef op = 1;
25 | */
26 | java.util.List
27 | getOpOrBuilderList();
28 | /**
29 | * repeated .tensorflow.ApiDef op = 1;
30 | */
31 | org.tensorflow.framework.ApiDefOrBuilder getOpOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/get_model_metadata.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow.serving;
4 | option cc_enable_arenas = true;
5 |
6 | import "google/protobuf/any.proto";
7 | import "tensorflow/core/protobuf/meta_graph.proto";
8 | import "tensorflow_serving/apis/model.proto";
9 |
10 | // Message returned for "signature_def" field.
11 | message SignatureDefMap {
12 | maprepeated .tensorflow.Feature feature = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.Feature feature = 1;
17 | */
18 | org.tensorflow.example.Feature getFeature(int index);
19 | /**
20 | * repeated .tensorflow.Feature feature = 1;
21 | */
22 | int getFeatureCount();
23 | /**
24 | * repeated .tensorflow.Feature feature = 1;
25 | */
26 | java.util.List
27 | getFeatureOrBuilderList();
28 | /**
29 | * repeated .tensorflow.Feature feature = 1;
30 | */
31 | org.tensorflow.example.FeatureOrBuilder getFeatureOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/model.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow.serving;
4 | option cc_enable_arenas = true;
5 |
6 | import "google/protobuf/wrappers.proto";
7 |
8 | // Metadata for an inference request such as the model name and version.
9 | message ModelSpec {
10 | // Required servable name.
11 | string name = 1;
12 |
13 | // Optional choice of which version of the model to use.
14 | //
15 | // Recommended to be left unset in the common case. Should be specified only
16 | // when there is a strong version consistency requirement.
17 | //
18 | // When left unspecified, the system will serve the best available version.
19 | // This is typically the latest version, though during version transitions,
20 | // notably when serving on a fleet of instances, may be either the previous or
21 | // new version.
22 | oneof version_choice {
23 | // Use this specific version number.
24 | google.protobuf.Int64Value version = 2;
25 |
26 | // Use the version associated with the given label.
27 | string version_label = 4;
28 | }
29 |
30 | // A named signature to evaluate. If unspecified, the default signature will
31 | // be used.
32 | string signature_name = 3;
33 | }
34 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/KernelListOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/kernel_def.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface KernelListOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.KernelList)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.KernelDef kernel = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.KernelDef kernel = 1;
17 | */
18 | org.tensorflow.framework.KernelDef getKernel(int index);
19 | /**
20 | * repeated .tensorflow.KernelDef kernel = 1;
21 | */
22 | int getKernelCount();
23 | /**
24 | * repeated .tensorflow.KernelDef kernel = 1;
25 | */
26 | java.util.List
27 | getKernelOrBuilderList();
28 | /**
29 | * repeated .tensorflow.KernelDef kernel = 1;
30 | */
31 | org.tensorflow.framework.KernelDefOrBuilder getKernelOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/GradientDefOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/function.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface GradientDefOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GradientDef)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The function name. 13 | *14 | * 15 | *
string function_name = 1;
16 | */
17 | java.lang.String getFunctionName();
18 | /**
19 | * 20 | * The function name. 21 | *22 | * 23 | *
string function_name = 1;
24 | */
25 | com.google.protobuf.ByteString
26 | getFunctionNameBytes();
27 |
28 | /**
29 | * 30 | * The gradient function's name. 31 | *32 | * 33 | *
string gradient_func = 2;
34 | */
35 | java.lang.String getGradientFunc();
36 | /**
37 | * 38 | * The gradient function's name. 39 | *40 | * 41 | *
string gradient_func = 2;
42 | */
43 | com.google.protobuf.ByteString
44 | getGradientFuncBytes();
45 | }
46 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/prediction_service.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow.serving;
4 | option cc_enable_arenas = true;
5 |
6 | import "tensorflow_serving/apis/classification.proto";
7 | import "tensorflow_serving/apis/get_model_metadata.proto";
8 | import "tensorflow_serving/apis/inference.proto";
9 | import "tensorflow_serving/apis/predict.proto";
10 | import "tensorflow_serving/apis/regression.proto";
11 |
12 | // open source marker; do not remove
13 | // PredictionService provides access to machine-learned models loaded by
14 | // model_servers.
15 | service PredictionService {
16 | // Classify.
17 | rpc Classify(ClassificationRequest) returns (ClassificationResponse);
18 |
19 | // Regress.
20 | rpc Regress(RegressionRequest) returns (RegressionResponse);
21 |
22 | // Predict -- provides access to loaded TensorFlow model.
23 | rpc Predict(PredictRequest) returns (PredictResponse);
24 |
25 | // MultiInference API for multi-headed models.
26 | rpc MultiInference(MultiInferenceRequest) returns (MultiInferenceResponse);
27 |
28 | // GetModelMetadata - provides access to metadata for loaded models.
29 | rpc GetModelMetadata(GetModelMetadataRequest)
30 | returns (GetModelMetadataResponse);
31 | }
32 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/SequenceExampleOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/example.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface SequenceExampleOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.SequenceExample)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * .tensorflow.Features context = 1;
12 | */
13 | boolean hasContext();
14 | /**
15 | * .tensorflow.Features context = 1;
16 | */
17 | org.tensorflow.example.Features getContext();
18 | /**
19 | * .tensorflow.Features context = 1;
20 | */
21 | org.tensorflow.example.FeaturesOrBuilder getContextOrBuilder();
22 |
23 | /**
24 | * .tensorflow.FeatureLists feature_lists = 2;
25 | */
26 | boolean hasFeatureLists();
27 | /**
28 | * .tensorflow.FeatureLists feature_lists = 2;
29 | */
30 | org.tensorflow.example.FeatureLists getFeatureLists();
31 | /**
32 | * .tensorflow.FeatureLists feature_lists = 2;
33 | */
34 | org.tensorflow.example.FeatureListsOrBuilder getFeatureListsOrBuilder();
35 | }
36 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/ReleaseCallableRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface ReleaseCallableRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ReleaseCallableRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * REQUIRED: session_handle must be returned by a CreateSession call 13 | * to the same master service. 14 | *15 | * 16 | *
string session_handle = 1;
17 | */
18 | java.lang.String getSessionHandle();
19 | /**
20 | * 21 | * REQUIRED: session_handle must be returned by a CreateSession call 22 | * to the same master service. 23 | *24 | * 25 | *
string session_handle = 1;
26 | */
27 | com.google.protobuf.ByteString
28 | getSessionHandleBytes();
29 |
30 | /**
31 | * 32 | * REQUIRED: handle must be returned by a MakeCallable call to the same 33 | * master service. 34 | *35 | * 36 | *
int64 handle = 2;
37 | */
38 | long getHandle();
39 | }
40 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/PartialRunSetupResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface PartialRunSetupResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.PartialRunSetupResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The unique handle corresponding to the ongoing partial run call setup by 13 | * the invocation to PartialRunSetup. This handle may be passed to 14 | * RunStepRequest to send and receive tensors for this partial run. 15 | *16 | * 17 | *
string partial_run_handle = 1;
18 | */
19 | java.lang.String getPartialRunHandle();
20 | /**
21 | * 22 | * The unique handle corresponding to the ongoing partial run call setup by 23 | * the invocation to PartialRunSetup. This handle may be passed to 24 | * RunStepRequest to send and receive tensors for this partial run. 25 | *26 | * 27 | *
string partial_run_handle = 1;
28 | */
29 | com.google.protobuf.ByteString
30 | getPartialRunHandleBytes();
31 | }
32 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/GraphTransferGraphInputNodeInfoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/graph_transfer_info.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface GraphTransferGraphInputNodeInfoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GraphTransferGraphInputNodeInfo)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * string name = 1;
12 | */
13 | java.lang.String getName();
14 | /**
15 | * string name = 1;
16 | */
17 | com.google.protobuf.ByteString
18 | getNameBytes();
19 |
20 | /**
21 | * repeated int64 shape = 2;
22 | */
23 | java.util.Listrepeated int64 shape = 2;
26 | */
27 | int getShapeCount();
28 | /**
29 | * repeated int64 shape = 2;
30 | */
31 | long getShape(int index);
32 |
33 | /**
34 | * .tensorflow.DataType dtype = 3;
35 | */
36 | int getDtypeValue();
37 | /**
38 | * .tensorflow.DataType dtype = 3;
39 | */
40 | org.tensorflow.framework.DataType getDtype();
41 | }
42 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/GraphTransferGraphOutputNodeInfoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/graph_transfer_info.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface GraphTransferGraphOutputNodeInfoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GraphTransferGraphOutputNodeInfo)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * string name = 1;
12 | */
13 | java.lang.String getName();
14 | /**
15 | * string name = 1;
16 | */
17 | com.google.protobuf.ByteString
18 | getNameBytes();
19 |
20 | /**
21 | * repeated int64 shape = 2;
22 | */
23 | java.util.Listrepeated int64 shape = 2;
26 | */
27 | int getShapeCount();
28 | /**
29 | * repeated int64 shape = 2;
30 | */
31 | long getShape(int index);
32 |
33 | /**
34 | * .tensorflow.DataType dtype = 3;
35 | */
36 | int getDtypeValue();
37 | /**
38 | * .tensorflow.DataType dtype = 3;
39 | */
40 | org.tensorflow.framework.DataType getDtype();
41 | }
42 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/LocalLinksOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/device_attributes.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface LocalLinksOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.LocalLinks)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.InterconnectLink link = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.InterconnectLink link = 1;
17 | */
18 | org.tensorflow.framework.InterconnectLink getLink(int index);
19 | /**
20 | * repeated .tensorflow.InterconnectLink link = 1;
21 | */
22 | int getLinkCount();
23 | /**
24 | * repeated .tensorflow.InterconnectLink link = 1;
25 | */
26 | java.util.List
27 | getLinkOrBuilderList();
28 | /**
29 | * repeated .tensorflow.InterconnectLink link = 1;
30 | */
31 | org.tensorflow.framework.InterconnectLinkOrBuilder getLinkOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/CostGraphDefOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/cost_graph.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface CostGraphDefOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CostGraphDef)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.CostGraphDef.Node node = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.CostGraphDef.Node node = 1;
17 | */
18 | org.tensorflow.framework.CostGraphDef.Node getNode(int index);
19 | /**
20 | * repeated .tensorflow.CostGraphDef.Node node = 1;
21 | */
22 | int getNodeCount();
23 | /**
24 | * repeated .tensorflow.CostGraphDef.Node node = 1;
25 | */
26 | java.util.List
27 | getNodeOrBuilderList();
28 | /**
29 | * repeated .tensorflow.CostGraphDef.Node node = 1;
30 | */
31 | org.tensorflow.framework.CostGraphDef.NodeOrBuilder getNodeOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/LoggingResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface LoggingResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.LoggingResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.LabeledStepStats step = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.LabeledStepStats step = 1;
17 | */
18 | org.tensorflow.distruntime.LabeledStepStats getStep(int index);
19 | /**
20 | * repeated .tensorflow.LabeledStepStats step = 1;
21 | */
22 | int getStepCount();
23 | /**
24 | * repeated .tensorflow.LabeledStepStats step = 1;
25 | */
26 | java.util.List
27 | getStepOrBuilderList();
28 | /**
29 | * repeated .tensorflow.LabeledStepStats step = 1;
30 | */
31 | org.tensorflow.distruntime.LabeledStepStatsOrBuilder getStepOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/CriticalSectionExecutionDefOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/critical_section.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface CriticalSectionExecutionDefOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CriticalSectionExecutionDef)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Name of the critical section handle. 13 | *14 | * 15 | *
string execute_in_critical_section_name = 1;
16 | */
17 | java.lang.String getExecuteInCriticalSectionName();
18 | /**
19 | * 20 | * Name of the critical section handle. 21 | *22 | * 23 | *
string execute_in_critical_section_name = 1;
24 | */
25 | com.google.protobuf.ByteString
26 | getExecuteInCriticalSectionNameBytes();
27 |
28 | /**
29 | * 30 | * Whether this operation requires exclusive access to its resources, 31 | * (i.e., no other CriticalSections may request the same resources). 32 | *33 | * 34 | *
bool exclusive_resource_access = 2;
35 | */
36 | boolean getExclusiveResourceAccess();
37 | }
38 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/StepStatsOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/step_stats.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface StepStatsOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.StepStats)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.DeviceStepStats dev_stats = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.DeviceStepStats dev_stats = 1;
17 | */
18 | org.tensorflow.framework.DeviceStepStats getDevStats(int index);
19 | /**
20 | * repeated .tensorflow.DeviceStepStats dev_stats = 1;
21 | */
22 | int getDevStatsCount();
23 | /**
24 | * repeated .tensorflow.DeviceStepStats dev_stats = 1;
25 | */
26 | java.util.List
27 | getDevStatsOrBuilderList();
28 | /**
29 | * repeated .tensorflow.DeviceStepStats dev_stats = 1;
30 | */
31 | org.tensorflow.framework.DeviceStepStatsOrBuilder getDevStatsOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/GetStepSequenceResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface GetStepSequenceResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GetStepSequenceResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.StepSequence step_sequence = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.StepSequence step_sequence = 1;
17 | */
18 | org.tensorflow.distruntime.StepSequence getStepSequence(int index);
19 | /**
20 | * repeated .tensorflow.StepSequence step_sequence = 1;
21 | */
22 | int getStepSequenceCount();
23 | /**
24 | * repeated .tensorflow.StepSequence step_sequence = 1;
25 | */
26 | java.util.List
27 | getStepSequenceOrBuilderList();
28 | /**
29 | * repeated .tensorflow.StepSequence step_sequence = 1;
30 | */
31 | org.tensorflow.distruntime.StepSequenceOrBuilder getStepSequenceOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/example/example_parser_configuration.proto:
--------------------------------------------------------------------------------
1 | // Protocol messages for describing the configuration of the ExampleParserOp.
2 |
3 | syntax = "proto3";
4 |
5 | option cc_enable_arenas = true;
6 | option java_outer_classname = "ExampleParserConfigurationProtos";
7 | option java_multiple_files = true;
8 | option java_package = "org.tensorflow.example";
9 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/example";
10 | package tensorflow;
11 |
12 | import "tensorflow/core/framework/tensor_shape.proto";
13 | import "tensorflow/core/framework/tensor.proto";
14 | import "tensorflow/core/framework/types.proto";
15 |
16 | message VarLenFeatureProto {
17 | tensorflow.DataType dtype = 1;
18 | string values_output_tensor_name = 2;
19 | string indices_output_tensor_name = 3;
20 | string shapes_output_tensor_name = 4;
21 | };
22 |
23 | message FixedLenFeatureProto {
24 | tensorflow.DataType dtype = 1;
25 | tensorflow.TensorShapeProto shape = 2;
26 | tensorflow.TensorProto default_value = 3;
27 | string values_output_tensor_name = 4;
28 | };
29 |
30 | message FeatureConfiguration {
31 | oneof config {
32 | FixedLenFeatureProto fixed_len_feature = 1;
33 | VarLenFeatureProto var_len_feature = 2;
34 | }
35 | };
36 |
37 | message ExampleParserConfiguration {
38 | map12 | * If present, only perform optimization for these ops. 13 | *14 | * 15 | *
repeated string enable_op = 1;
16 | */
17 | java.util.List21 | * If present, only perform optimization for these ops. 22 | *23 | * 24 | *
repeated string enable_op = 1;
25 | */
26 | int getEnableOpCount();
27 | /**
28 | * 29 | * If present, only perform optimization for these ops. 30 | *31 | * 32 | *
repeated string enable_op = 1;
33 | */
34 | java.lang.String getEnableOp(int index);
35 | /**
36 | * 37 | * If present, only perform optimization for these ops. 38 | *39 | * 40 | *
repeated string enable_op = 1;
41 | */
42 | com.google.protobuf.ByteString
43 | getEnableOpBytes(int index);
44 | }
45 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/ControlFlowContextDefOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/control_flow.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface ControlFlowContextDefOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ControlFlowContextDef)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * .tensorflow.CondContextDef cond_ctxt = 1;
12 | */
13 | boolean hasCondCtxt();
14 | /**
15 | * .tensorflow.CondContextDef cond_ctxt = 1;
16 | */
17 | org.tensorflow.framework.CondContextDef getCondCtxt();
18 | /**
19 | * .tensorflow.CondContextDef cond_ctxt = 1;
20 | */
21 | org.tensorflow.framework.CondContextDefOrBuilder getCondCtxtOrBuilder();
22 |
23 | /**
24 | * .tensorflow.WhileContextDef while_ctxt = 2;
25 | */
26 | boolean hasWhileCtxt();
27 | /**
28 | * .tensorflow.WhileContextDef while_ctxt = 2;
29 | */
30 | org.tensorflow.framework.WhileContextDef getWhileCtxt();
31 | /**
32 | * .tensorflow.WhileContextDef while_ctxt = 2;
33 | */
34 | org.tensorflow.framework.WhileContextDefOrBuilder getWhileCtxtOrBuilder();
35 |
36 | public org.tensorflow.framework.ControlFlowContextDef.CtxtCase getCtxtCase();
37 | }
38 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/GraphTransferNodeInfoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/graph_transfer_info.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface GraphTransferNodeInfoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GraphTransferNodeInfo)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * string name = 1;
12 | */
13 | java.lang.String getName();
14 | /**
15 | * string name = 1;
16 | */
17 | com.google.protobuf.ByteString
18 | getNameBytes();
19 |
20 | /**
21 | * int32 node_id = 2;
22 | */
23 | int getNodeId();
24 |
25 | /**
26 | * string type_name = 3;
27 | */
28 | java.lang.String getTypeName();
29 | /**
30 | * string type_name = 3;
31 | */
32 | com.google.protobuf.ByteString
33 | getTypeNameBytes();
34 |
35 | /**
36 | * int32 soc_op_id = 4;
37 | */
38 | int getSocOpId();
39 |
40 | /**
41 | * int32 padding_id = 5;
42 | */
43 | int getPaddingId();
44 |
45 | /**
46 | * int32 input_count = 6;
47 | */
48 | int getInputCount();
49 |
50 | /**
51 | * int32 output_count = 7;
52 | */
53 | int getOutputCount();
54 | }
55 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/GetStatusResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface GetStatusResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GetStatusResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.DeviceAttributes device_attributes = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.DeviceAttributes device_attributes = 1;
17 | */
18 | org.tensorflow.framework.DeviceAttributes getDeviceAttributes(int index);
19 | /**
20 | * repeated .tensorflow.DeviceAttributes device_attributes = 1;
21 | */
22 | int getDeviceAttributesCount();
23 | /**
24 | * repeated .tensorflow.DeviceAttributes device_attributes = 1;
25 | */
26 | java.util.List
27 | getDeviceAttributesOrBuilderList();
28 | /**
29 | * repeated .tensorflow.DeviceAttributes device_attributes = 1;
30 | */
31 | org.tensorflow.framework.DeviceAttributesOrBuilder getDeviceAttributesOrBuilder(
32 | int index);
33 | }
34 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/prediction_log.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | option cc_enable_arenas = true;
4 |
5 | import "tensorflow_serving/apis/classification.proto";
6 | import "tensorflow_serving/apis/inference.proto";
7 | import "tensorflow_serving/apis/predict.proto";
8 | import "tensorflow_serving/apis/regression.proto";
9 | import "tensorflow_serving/apis/session_service.proto";
10 | import "tensorflow_serving/core/logging.proto";
11 |
12 | package tensorflow.serving;
13 |
14 | message ClassifyLog {
15 | ClassificationRequest request = 1;
16 | ClassificationResponse response = 2;
17 | }
18 |
19 | message RegressLog {
20 | RegressionRequest request = 1;
21 | RegressionResponse response = 2;
22 | }
23 |
24 | message PredictLog {
25 | PredictRequest request = 1;
26 | PredictResponse response = 2;
27 | }
28 |
29 | message MultiInferenceLog {
30 | MultiInferenceRequest request = 1;
31 | MultiInferenceResponse response = 2;
32 | }
33 |
34 | message SessionRunLog {
35 | SessionRunRequest request = 1;
36 | SessionRunResponse response = 2;
37 | }
38 |
39 | // Logged model inference request.
40 | message PredictionLog {
41 | LogMetadata log_metadata = 1;
42 | oneof log_type {
43 | ClassifyLog classify_log = 2;
44 | RegressLog regress_log = 3;
45 | PredictLog predict_log = 6;
46 | MultiInferenceLog multi_inference_log = 4;
47 | SessionRunLog session_run_log = 5;
48 | }
49 | }
50 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/CompleteGroupRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface CompleteGroupRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CompleteGroupRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int32 group_key = 1;
12 | */
13 | int getGroupKey();
14 |
15 | /**
16 | * int32 group_size = 2;
17 | */
18 | int getGroupSize();
19 |
20 | /**
21 | * string device_type = 3;
22 | */
23 | java.lang.String getDeviceType();
24 | /**
25 | * string device_type = 3;
26 | */
27 | com.google.protobuf.ByteString
28 | getDeviceTypeBytes();
29 |
30 | /**
31 | * repeated string device_name = 4;
32 | */
33 | java.util.Listrepeated string device_name = 4;
37 | */
38 | int getDeviceNameCount();
39 | /**
40 | * repeated string device_name = 4;
41 | */
42 | java.lang.String getDeviceName(int index);
43 | /**
44 | * repeated string device_name = 4;
45 | */
46 | com.google.protobuf.ByteString
47 | getDeviceNameBytes(int index);
48 | }
49 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/variable.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "VariableProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 |
10 | // Protocol buffer representing a Variable.
11 | message VariableDef {
12 | // Name of the variable tensor.
13 | string variable_name = 1;
14 |
15 | // Name of the tensor holding the variable's initial value.
16 | string initial_value_name = 6;
17 |
18 | // Name of the initializer op.
19 | string initializer_name = 2;
20 |
21 | // Name of the snapshot tensor.
22 | string snapshot_name = 3;
23 |
24 | // Support for saving variables as slices of a larger variable.
25 | SaveSliceInfoDef save_slice_info_def = 4;
26 |
27 | // Whether to represent this as a ResourceVariable.
28 | bool is_resource = 5;
29 |
30 | // Whether this variable should be trained.
31 | bool trainable = 7;
32 | }
33 |
34 | message SaveSliceInfoDef {
35 | // Name of the full variable of which this is a slice.
36 | string full_name = 1;
37 | // Shape of the full variable.
38 | repeated int64 full_shape = 2;
39 | // Offset of this variable into the full variable.
40 | repeated int64 var_offset = 3;
41 | // Shape of this variable.
42 | repeated int64 var_shape = 4;
43 | }
44 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/CreateSessionResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface CreateSessionResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CreateSessionResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The session handle to be used in subsequent calls for the created session. 13 | * The client must arrange to call CloseSession with this returned 14 | * session handle to close the session. 15 | *16 | * 17 | *
string session_handle = 1;
18 | */
19 | java.lang.String getSessionHandle();
20 | /**
21 | * 22 | * The session handle to be used in subsequent calls for the created session. 23 | * The client must arrange to call CloseSession with this returned 24 | * session handle to close the session. 25 | *26 | * 27 | *
string session_handle = 1;
28 | */
29 | com.google.protobuf.ByteString
30 | getSessionHandleBytes();
31 |
32 | /**
33 | * 34 | * The initial version number for the graph, to be used in the next call 35 | * to ExtendSession. 36 | *37 | * 38 | *
int64 graph_version = 2;
39 | */
40 | long getGraphVersion();
41 | }
42 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/classification.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | option cc_enable_arenas = true;
4 |
5 | import "tensorflow_serving/apis/input.proto";
6 | import "tensorflow_serving/apis/model.proto";
7 |
8 | package tensorflow.serving;
9 |
10 | // A single class.
11 | message Class {
12 | // Label or name of the class.
13 | string label = 1;
14 | // Score for this class (e.g., the probability the item belongs to this
15 | // class). As per the proto3 default-value semantics, if the score is missing,
16 | // it should be treated as 0.
17 | float score = 2;
18 | }
19 |
20 | // List of classes for a single item (tensorflow.Example).
21 | message Classifications {
22 | repeated Class classes = 1;
23 | }
24 |
25 | // Contains one result per input example, in the same order as the input in
26 | // ClassificationRequest.
27 | message ClassificationResult {
28 | repeated Classifications classifications = 1;
29 | }
30 |
31 | // RPC Interfaces
32 |
33 | message ClassificationRequest {
34 | // Model Specification. If version is not specified, will use the latest
35 | // (numerical) version.
36 | ModelSpec model_spec = 1;
37 |
38 | // Input data.
39 | tensorflow.serving.Input input = 2;
40 | }
41 |
42 | message ClassificationResponse {
43 | // Effective Model Specification used for classification.
44 | ModelSpec model_spec = 2;
45 |
46 | // Result of the classification.
47 | ClassificationResult result = 1;
48 | }
49 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/GraphTransferConstNodeInfoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/graph_transfer_info.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface GraphTransferConstNodeInfoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GraphTransferConstNodeInfo)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * string name = 1;
12 | */
13 | java.lang.String getName();
14 | /**
15 | * string name = 1;
16 | */
17 | com.google.protobuf.ByteString
18 | getNameBytes();
19 |
20 | /**
21 | * int32 node_id = 2;
22 | */
23 | int getNodeId();
24 |
25 | /**
26 | * repeated int64 shape = 3;
27 | */
28 | java.util.Listrepeated int64 shape = 3;
31 | */
32 | int getShapeCount();
33 | /**
34 | * repeated int64 shape = 3;
35 | */
36 | long getShape(int index);
37 |
38 | /**
39 | * bytes data = 4;
40 | */
41 | com.google.protobuf.ByteString getData();
42 |
43 | /**
44 | * .tensorflow.DataType dtype = 5;
45 | */
46 | int getDtypeValue();
47 | /**
48 | * .tensorflow.DataType dtype = 5;
49 | */
50 | org.tensorflow.framework.DataType getDtype();
51 | }
52 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/tensor_slice.proto:
--------------------------------------------------------------------------------
1 | // Protocol buffer representing slices of a tensor
2 |
3 | syntax = "proto3";
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "TensorSliceProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 |
10 | package tensorflow;
11 |
12 | // Can only be interpreted if you know the corresponding TensorShape.
13 | message TensorSliceProto {
14 | // Extent of the slice in one dimension.
15 | message Extent {
16 | // Either both or no attributes must be set. When no attribute is set
17 | // means: All data in that dimension.
18 |
19 | // Start index of the slice, starting at 0.
20 | int64 start = 1;
21 |
22 | // Length of the slice: if the length is missing or -1 we will
23 | // interpret this as "everything in this dimension". We use
24 | // "oneof" to preserve information about whether the length is
25 | // present without changing the serialization format from the
26 | // prior proto2 version of this proto.
27 | oneof has_length {
28 | int64 length = 2;
29 | }
30 | };
31 |
32 | // Extent of the slice in all tensor dimensions.
33 | //
34 | // Must have one entry for each of the dimension of the tensor that this
35 | // slice belongs to. The order of sizes is the same as the order of
36 | // dimensions in the TensorShape.
37 | repeated Extent extent = 1;
38 | };
39 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/FeatureConfigurationOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/example_parser_configuration.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface FeatureConfigurationOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.FeatureConfiguration)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * .tensorflow.FixedLenFeatureProto fixed_len_feature = 1;
12 | */
13 | boolean hasFixedLenFeature();
14 | /**
15 | * .tensorflow.FixedLenFeatureProto fixed_len_feature = 1;
16 | */
17 | org.tensorflow.example.FixedLenFeatureProto getFixedLenFeature();
18 | /**
19 | * .tensorflow.FixedLenFeatureProto fixed_len_feature = 1;
20 | */
21 | org.tensorflow.example.FixedLenFeatureProtoOrBuilder getFixedLenFeatureOrBuilder();
22 |
23 | /**
24 | * .tensorflow.VarLenFeatureProto var_len_feature = 2;
25 | */
26 | boolean hasVarLenFeature();
27 | /**
28 | * .tensorflow.VarLenFeatureProto var_len_feature = 2;
29 | */
30 | org.tensorflow.example.VarLenFeatureProto getVarLenFeature();
31 | /**
32 | * .tensorflow.VarLenFeatureProto var_len_feature = 2;
33 | */
34 | org.tensorflow.example.VarLenFeatureProtoOrBuilder getVarLenFeatureOrBuilder();
35 |
36 | public org.tensorflow.example.FeatureConfiguration.ConfigCase getConfigCase();
37 | }
38 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/GraphTransferNodeInputInfoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/graph_transfer_info.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface GraphTransferNodeInputInfoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.GraphTransferNodeInputInfo)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int32 node_id = 1;
12 | */
13 | int getNodeId();
14 |
15 | /**
16 | * repeated .tensorflow.GraphTransferNodeInput node_input = 2;
17 | */
18 | java.util.Listrepeated .tensorflow.GraphTransferNodeInput node_input = 2;
22 | */
23 | org.tensorflow.framework.GraphTransferNodeInput getNodeInput(int index);
24 | /**
25 | * repeated .tensorflow.GraphTransferNodeInput node_input = 2;
26 | */
27 | int getNodeInputCount();
28 | /**
29 | * repeated .tensorflow.GraphTransferNodeInput node_input = 2;
30 | */
31 | java.util.List
32 | getNodeInputOrBuilderList();
33 | /**
34 | * repeated .tensorflow.GraphTransferNodeInput node_input = 2;
35 | */
36 | org.tensorflow.framework.GraphTransferNodeInputOrBuilder getNodeInputOrBuilder(
37 | int index);
38 | }
39 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/DeviceStepStatsOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/step_stats.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface DeviceStepStatsOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.DeviceStepStats)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * string device = 1;
12 | */
13 | java.lang.String getDevice();
14 | /**
15 | * string device = 1;
16 | */
17 | com.google.protobuf.ByteString
18 | getDeviceBytes();
19 |
20 | /**
21 | * repeated .tensorflow.NodeExecStats node_stats = 2;
22 | */
23 | java.util.Listrepeated .tensorflow.NodeExecStats node_stats = 2;
27 | */
28 | org.tensorflow.framework.NodeExecStats getNodeStats(int index);
29 | /**
30 | * repeated .tensorflow.NodeExecStats node_stats = 2;
31 | */
32 | int getNodeStatsCount();
33 | /**
34 | * repeated .tensorflow.NodeExecStats node_stats = 2;
35 | */
36 | java.util.List
37 | getNodeStatsOrBuilderList();
38 | /**
39 | * repeated .tensorflow.NodeExecStats node_stats = 2;
40 | */
41 | org.tensorflow.framework.NodeExecStatsOrBuilder getNodeStatsOrBuilder(
42 | int index);
43 | }
44 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/VersionDefOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/versions.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface VersionDefOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.VersionDef)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The version of the code that produced this data. 13 | *14 | * 15 | *
int32 producer = 1;
16 | */
17 | int getProducer();
18 |
19 | /**
20 | * 21 | * Any consumer below this version is not allowed to consume this data. 22 | *23 | * 24 | *
int32 min_consumer = 2;
25 | */
26 | int getMinConsumer();
27 |
28 | /**
29 | * 30 | * Specific consumer versions which are disallowed (e.g. due to bugs). 31 | *32 | * 33 | *
repeated int32 bad_consumers = 3;
34 | */
35 | java.util.List38 | * Specific consumer versions which are disallowed (e.g. due to bugs). 39 | *40 | * 41 | *
repeated int32 bad_consumers = 3;
42 | */
43 | int getBadConsumersCount();
44 | /**
45 | * 46 | * Specific consumer versions which are disallowed (e.g. due to bugs). 47 | *48 | * 49 | *
repeated int32 bad_consumers = 3;
50 | */
51 | int getBadConsumers(int index);
52 | }
53 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/TensorConnectionOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/config.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface TensorConnectionOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.TensorConnection)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * A tensor name. The value of this tensor will be substituted for 13 | * the tensor named in `to_tensor`. 14 | *15 | * 16 | *
string from_tensor = 1;
17 | */
18 | java.lang.String getFromTensor();
19 | /**
20 | * 21 | * A tensor name. The value of this tensor will be substituted for 22 | * the tensor named in `to_tensor`. 23 | *24 | * 25 | *
string from_tensor = 1;
26 | */
27 | com.google.protobuf.ByteString
28 | getFromTensorBytes();
29 |
30 | /**
31 | * 32 | * A tensor name. The value of this tensor will be bound to the 33 | * value of the tensor named in `from_tensor`. 34 | *35 | * 36 | *
string to_tensor = 2;
37 | */
38 | java.lang.String getToTensor();
39 | /**
40 | * 41 | * A tensor name. The value of this tensor will be bound to the 42 | * value of the tensor named in `from_tensor`. 43 | *44 | * 45 | *
string to_tensor = 2;
46 | */
47 | com.google.protobuf.ByteString
48 | getToTensorBytes();
49 | }
50 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/DeviceLocalityOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/device_attributes.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface DeviceLocalityOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.DeviceLocality)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Optional bus locality of device. Default value of 0 means 13 | * no specific locality. Specific localities are indexed from 1. 14 | *15 | * 16 | *
int32 bus_id = 1;
17 | */
18 | int getBusId();
19 |
20 | /**
21 | * 22 | * Optional NUMA locality of device. 23 | *24 | * 25 | *
int32 numa_node = 2;
26 | */
27 | int getNumaNode();
28 |
29 | /**
30 | * 31 | * Optional local interconnect links to other devices. 32 | *33 | * 34 | *
.tensorflow.LocalLinks links = 3;
35 | */
36 | boolean hasLinks();
37 | /**
38 | * 39 | * Optional local interconnect links to other devices. 40 | *41 | * 42 | *
.tensorflow.LocalLinks links = 3;
43 | */
44 | org.tensorflow.framework.LocalLinks getLinks();
45 | /**
46 | * 47 | * Optional local interconnect links to other devices. 48 | *49 | * 50 | *
.tensorflow.LocalLinks links = 3;
51 | */
52 | org.tensorflow.framework.LocalLinksOrBuilder getLinksOrBuilder();
53 | }
54 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/VarLenFeatureProtoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/example_parser_configuration.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface VarLenFeatureProtoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.VarLenFeatureProto)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * .tensorflow.DataType dtype = 1;
12 | */
13 | int getDtypeValue();
14 | /**
15 | * .tensorflow.DataType dtype = 1;
16 | */
17 | org.tensorflow.framework.DataType getDtype();
18 |
19 | /**
20 | * string values_output_tensor_name = 2;
21 | */
22 | java.lang.String getValuesOutputTensorName();
23 | /**
24 | * string values_output_tensor_name = 2;
25 | */
26 | com.google.protobuf.ByteString
27 | getValuesOutputTensorNameBytes();
28 |
29 | /**
30 | * string indices_output_tensor_name = 3;
31 | */
32 | java.lang.String getIndicesOutputTensorName();
33 | /**
34 | * string indices_output_tensor_name = 3;
35 | */
36 | com.google.protobuf.ByteString
37 | getIndicesOutputTensorNameBytes();
38 |
39 | /**
40 | * string shapes_output_tensor_name = 4;
41 | */
42 | java.lang.String getShapesOutputTensorName();
43 | /**
44 | * string shapes_output_tensor_name = 4;
45 | */
46 | com.google.protobuf.ByteString
47 | getShapesOutputTensorNameBytes();
48 | }
49 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/util/BundleHeaderProtoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/tensor_bundle.proto
3 |
4 | package org.tensorflow.util;
5 |
6 | public interface BundleHeaderProtoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.BundleHeaderProto)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Number of data files in the bundle. 13 | *14 | * 15 | *
int32 num_shards = 1;
16 | */
17 | int getNumShards();
18 |
19 | /**
20 | * .tensorflow.BundleHeaderProto.Endianness endianness = 2;
21 | */
22 | int getEndiannessValue();
23 | /**
24 | * .tensorflow.BundleHeaderProto.Endianness endianness = 2;
25 | */
26 | org.tensorflow.util.BundleHeaderProto.Endianness getEndianness();
27 |
28 | /**
29 | * 30 | * Versioning of the tensor bundle format. 31 | *32 | * 33 | *
.tensorflow.VersionDef version = 3;
34 | */
35 | boolean hasVersion();
36 | /**
37 | * 38 | * Versioning of the tensor bundle format. 39 | *40 | * 41 | *
.tensorflow.VersionDef version = 3;
42 | */
43 | org.tensorflow.framework.VersionDef getVersion();
44 | /**
45 | * 46 | * Versioning of the tensor bundle format. 47 | *48 | * 49 | *
.tensorflow.VersionDef version = 3;
50 | */
51 | org.tensorflow.framework.VersionDefOrBuilder getVersionOrBuilder();
52 | }
53 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/ClusterDefOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/cluster.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface ClusterDefOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ClusterDef)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The jobs that comprise the cluster. 13 | *14 | * 15 | *
repeated .tensorflow.JobDef job = 1;
16 | */
17 | java.util.List21 | * The jobs that comprise the cluster. 22 | *23 | * 24 | *
repeated .tensorflow.JobDef job = 1;
25 | */
26 | org.tensorflow.distruntime.JobDef getJob(int index);
27 | /**
28 | * 29 | * The jobs that comprise the cluster. 30 | *31 | * 32 | *
repeated .tensorflow.JobDef job = 1;
33 | */
34 | int getJobCount();
35 | /**
36 | * 37 | * The jobs that comprise the cluster. 38 | *39 | * 40 | *
repeated .tensorflow.JobDef job = 1;
41 | */
42 | java.util.List
43 | getJobOrBuilderList();
44 | /**
45 | * 46 | * The jobs that comprise the cluster. 47 | *48 | * 49 | *
repeated .tensorflow.JobDef job = 1;
50 | */
51 | org.tensorflow.distruntime.JobDefOrBuilder getJobOrBuilder(
52 | int index);
53 | }
54 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/predict.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow.serving;
4 | option cc_enable_arenas = true;
5 |
6 | import "tensorflow/core/framework/tensor.proto";
7 | import "tensorflow_serving/apis/model.proto";
8 |
9 | // PredictRequest specifies which TensorFlow model to run, as well as
10 | // how inputs are mapped to tensors and how outputs are filtered before
11 | // returning to user.
12 | message PredictRequest {
13 | // Model Specification. If version is not specified, will use the latest
14 | // (numerical) version.
15 | ModelSpec model_spec = 1;
16 |
17 | // Input tensors.
18 | // Names of input tensor are alias names. The mapping from aliases to real
19 | // input tensor names is stored in the SavedModel export as a prediction
20 | // SignatureDef under the 'inputs' field.
21 | map12 | * If true, always use RPC to contact the session target. 13 | * If false (the default option), TensorFlow may use an optimized 14 | * transport for client-master communication that avoids the RPC 15 | * stack. This option is primarily for used testing the RPC stack. 16 | *17 | * 18 | *
bool use_rpc_for_inprocess_master = 1;
19 | */
20 | boolean getUseRpcForInprocessMaster();
21 |
22 | /**
23 | * 24 | * The compression algorithm to be used. One of "deflate", "gzip". 25 | *26 | * 27 | *
string compression_algorithm = 2;
28 | */
29 | java.lang.String getCompressionAlgorithm();
30 | /**
31 | * 32 | * The compression algorithm to be used. One of "deflate", "gzip". 33 | *34 | * 35 | *
string compression_algorithm = 2;
36 | */
37 | com.google.protobuf.ByteString
38 | getCompressionAlgorithmBytes();
39 |
40 | /**
41 | * 42 | * If compression_algorithm is set, the compression level to be used. 43 | * From 0 (no compression), up to 3. 44 | *45 | * 46 | *
int32 compression_level = 3;
47 | */
48 | int getCompressionLevel();
49 | }
50 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/SummaryOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/summary.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface SummaryOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.Summary)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Set of values for the summary. 13 | *14 | * 15 | *
repeated .tensorflow.Summary.Value value = 1;
16 | */
17 | java.util.List21 | * Set of values for the summary. 22 | *23 | * 24 | *
repeated .tensorflow.Summary.Value value = 1;
25 | */
26 | org.tensorflow.framework.Summary.Value getValue(int index);
27 | /**
28 | * 29 | * Set of values for the summary. 30 | *31 | * 32 | *
repeated .tensorflow.Summary.Value value = 1;
33 | */
34 | int getValueCount();
35 | /**
36 | * 37 | * Set of values for the summary. 38 | *39 | * 40 | *
repeated .tensorflow.Summary.Value value = 1;
41 | */
42 | java.util.List
43 | getValueOrBuilderList();
44 | /**
45 | * 46 | * Set of values for the summary. 47 | *48 | * 49 | *
repeated .tensorflow.Summary.Value value = 1;
50 | */
51 | org.tensorflow.framework.Summary.ValueOrBuilder getValueOrBuilder(
52 | int index);
53 | }
54 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/kernel_def.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "KernelDefProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 | import "tensorflow/core/framework/attr_value.proto";
10 |
11 | message KernelDef {
12 | // Must match the name of an Op.
13 | string op = 1;
14 |
15 | // Type of device this kernel runs on.
16 | string device_type = 2;
17 |
18 | message AttrConstraint {
19 | // Name of an attr from the Op.
20 | string name = 1;
21 |
22 | // A list of values that this kernel supports for this attr.
23 | // Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
24 | AttrValue allowed_values = 2;
25 | }
26 | repeated AttrConstraint constraint = 3;
27 |
28 | // Names of the Op's input_/output_args that reside in host memory
29 | // instead of device memory.
30 | repeated string host_memory_arg = 4;
31 |
32 | // This allows experimental kernels to be registered for an op that
33 | // won't be used unless the user specifies a "_kernel" attr with
34 | // value matching this.
35 | string label = 5;
36 |
37 | // Prioritization of kernel amongst different devices. By default we assume
38 | // priority is 0. The higher the priority the better. By default (i.e. if
39 | // this is not set), we prefer GPU kernels over CPU.
40 | int32 priority = 6;
41 | }
42 |
43 | // A collection of KernelDefs
44 | message KernelList {
45 | repeated KernelDef kernel = 1;
46 | };
47 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/device_attributes.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "DeviceAttributesProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 |
10 | message InterconnectLink {
11 | int32 device_id = 1;
12 | string type = 2;
13 | int32 strength = 3;
14 | };
15 |
16 | message LocalLinks {
17 | repeated InterconnectLink link = 1;
18 | };
19 |
20 | message DeviceLocality {
21 | // Optional bus locality of device. Default value of 0 means
22 | // no specific locality. Specific localities are indexed from 1.
23 | int32 bus_id = 1;
24 |
25 | // Optional NUMA locality of device.
26 | int32 numa_node = 2;
27 |
28 | // Optional local interconnect links to other devices.
29 | LocalLinks links = 3;
30 | };
31 |
32 | message DeviceAttributes {
33 | // Fully specified name of the device within a cluster.
34 | string name = 1;
35 |
36 | // String representation of device_type.
37 | string device_type = 2;
38 |
39 | // Memory capacity of device in bytes.
40 | int64 memory_limit = 4;
41 |
42 | // Platform-specific data about device that may be useful
43 | // for supporting efficient data transfers.
44 | DeviceLocality locality = 5;
45 |
46 | // A device is assigned a global unique number each time it is
47 | // initialized. "incarnation" should never be 0.
48 | fixed64 incarnation = 6;
49 |
50 | // String representation of the physical device that this device maps to.
51 | string physical_device_desc = 7;
52 | }
53 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/FeatureOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/feature.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface FeatureOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.Feature)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * .tensorflow.BytesList bytes_list = 1;
12 | */
13 | boolean hasBytesList();
14 | /**
15 | * .tensorflow.BytesList bytes_list = 1;
16 | */
17 | org.tensorflow.example.BytesList getBytesList();
18 | /**
19 | * .tensorflow.BytesList bytes_list = 1;
20 | */
21 | org.tensorflow.example.BytesListOrBuilder getBytesListOrBuilder();
22 |
23 | /**
24 | * .tensorflow.FloatList float_list = 2;
25 | */
26 | boolean hasFloatList();
27 | /**
28 | * .tensorflow.FloatList float_list = 2;
29 | */
30 | org.tensorflow.example.FloatList getFloatList();
31 | /**
32 | * .tensorflow.FloatList float_list = 2;
33 | */
34 | org.tensorflow.example.FloatListOrBuilder getFloatListOrBuilder();
35 |
36 | /**
37 | * .tensorflow.Int64List int64_list = 3;
38 | */
39 | boolean hasInt64List();
40 | /**
41 | * .tensorflow.Int64List int64_list = 3;
42 | */
43 | org.tensorflow.example.Int64List getInt64List();
44 | /**
45 | * .tensorflow.Int64List int64_list = 3;
46 | */
47 | org.tensorflow.example.Int64ListOrBuilder getInt64ListOrBuilder();
48 |
49 | public org.tensorflow.example.Feature.KindCase getKindCase();
50 | }
51 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/util/IteratorStateMetadataOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/iterator.proto
3 |
4 | package org.tensorflow.util;
5 |
6 | public interface IteratorStateMetadataOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.IteratorStateMetadata)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * A user-specified version string. 13 | *14 | * 15 | *
string version = 1;
16 | */
17 | java.lang.String getVersion();
18 | /**
19 | * 20 | * A user-specified version string. 21 | *22 | * 23 | *
string version = 1;
24 | */
25 | com.google.protobuf.ByteString
26 | getVersionBytes();
27 |
28 | /**
29 | * 30 | * Keys for tensors in the VariantTensorDataProto. 31 | *32 | * 33 | *
repeated string keys = 2;
34 | */
35 | java.util.List39 | * Keys for tensors in the VariantTensorDataProto. 40 | *41 | * 42 | *
repeated string keys = 2;
43 | */
44 | int getKeysCount();
45 | /**
46 | * 47 | * Keys for tensors in the VariantTensorDataProto. 48 | *49 | * 50 | *
repeated string keys = 2;
51 | */
52 | java.lang.String getKeys(int index);
53 | /**
54 | * 55 | * Keys for tensors in the VariantTensorDataProto. 56 | *57 | * 58 | *
repeated string keys = 2;
59 | */
60 | com.google.protobuf.ByteString
61 | getKeysBytes(int index);
62 | }
63 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/internal/serialized_input.proto:
--------------------------------------------------------------------------------
1 | /* Copyright 2017 Google Inc. 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 | // Serialized counterparts of the messages in input.proto. These protos enable
17 | // us to keep the original tensorflow.serving.Input's structure but with the
18 | // tensorflow.Examples in their serialized form. When combined with lazy
19 | // parsing, this improves performance by allowing us to skip a redundant
20 | // deserialization/serialization loop.
21 | //
22 | // WARNING: These are internal implementation details and not part of the public
23 | // API.
24 |
25 | syntax = "proto3";
26 |
27 | option cc_enable_arenas = true;
28 |
29 | package tensorflow.serving.internal;
30 |
31 | message SerializedExampleList {
32 | repeated bytes examples = 1;
33 | }
34 |
35 | message SerializedExampleListWithContext {
36 | repeated bytes examples = 1;
37 | bytes context = 2;
38 | }
39 |
40 | message SerializedInput {
41 | oneof kind {
42 | SerializedExampleList example_list = 1;
43 | SerializedExampleListWithContext example_list_with_context = 2;
44 | }
45 | }
46 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/ExampleParserConfigurationOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/example_parser_configuration.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface ExampleParserConfigurationOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ExampleParserConfiguration)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * map<string, .tensorflow.FeatureConfiguration> feature_map = 1;
12 | */
13 | int getFeatureMapCount();
14 | /**
15 | * map<string, .tensorflow.FeatureConfiguration> feature_map = 1;
16 | */
17 | boolean containsFeatureMap(
18 | java.lang.String key);
19 | /**
20 | * Use {@link #getFeatureMapMap()} instead.
21 | */
22 | @java.lang.Deprecated
23 | java.util.Mapmap<string, .tensorflow.FeatureConfiguration> feature_map = 1;
27 | */
28 | java.util.Mapmap<string, .tensorflow.FeatureConfiguration> feature_map = 1;
32 | */
33 |
34 | org.tensorflow.example.FeatureConfiguration getFeatureMapOrDefault(
35 | java.lang.String key,
36 | org.tensorflow.example.FeatureConfiguration defaultValue);
37 | /**
38 | * map<string, .tensorflow.FeatureConfiguration> feature_map = 1;
39 | */
40 |
41 | org.tensorflow.example.FeatureConfiguration getFeatureMapOrThrow(
42 | java.lang.String key);
43 | }
44 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/NameAttrListOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/attr_value.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface NameAttrListOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.NameAttrList)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * string name = 1;
12 | */
13 | java.lang.String getName();
14 | /**
15 | * string name = 1;
16 | */
17 | com.google.protobuf.ByteString
18 | getNameBytes();
19 |
20 | /**
21 | * map<string, .tensorflow.AttrValue> attr = 2;
22 | */
23 | int getAttrCount();
24 | /**
25 | * map<string, .tensorflow.AttrValue> attr = 2;
26 | */
27 | boolean containsAttr(
28 | java.lang.String key);
29 | /**
30 | * Use {@link #getAttrMap()} instead.
31 | */
32 | @java.lang.Deprecated
33 | java.util.Mapmap<string, .tensorflow.AttrValue> attr = 2;
37 | */
38 | java.util.Mapmap<string, .tensorflow.AttrValue> attr = 2;
42 | */
43 |
44 | org.tensorflow.framework.AttrValue getAttrOrDefault(
45 | java.lang.String key,
46 | org.tensorflow.framework.AttrValue defaultValue);
47 | /**
48 | * map<string, .tensorflow.AttrValue> attr = 2;
49 | */
50 |
51 | org.tensorflow.framework.AttrValue getAttrOrThrow(
52 | java.lang.String key);
53 | }
54 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/ListDevicesRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface ListDevicesRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ListDevicesRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Optional: session_handle must be returned by a CreateSession call to the 13 | * same master service. 14 | * When session_handle is empty, the ClusterSpec provided when the master was 15 | * started is used to compute the available devices. If the session_handle is 16 | * provided but not recognized, an error is returned. Finally, if a valid 17 | * session_handle is provided, the cluster configuration for that session is 18 | * used when computing the response. 19 | *20 | * 21 | *
string session_handle = 1;
22 | */
23 | java.lang.String getSessionHandle();
24 | /**
25 | * 26 | * Optional: session_handle must be returned by a CreateSession call to the 27 | * same master service. 28 | * When session_handle is empty, the ClusterSpec provided when the master was 29 | * started is used to compute the available devices. If the session_handle is 30 | * provided but not recognized, an error is returned. Finally, if a valid 31 | * session_handle is provided, the cluster configuration for that session is 32 | * used when computing the response. 33 | *34 | * 35 | *
string session_handle = 1;
36 | */
37 | com.google.protobuf.ByteString
38 | getSessionHandleBytes();
39 | }
40 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/MakeCallableRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface MakeCallableRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.MakeCallableRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * REQUIRED: session_handle must be returned by a CreateSession call 13 | * to the same master service. 14 | *15 | * 16 | *
string session_handle = 1;
17 | */
18 | java.lang.String getSessionHandle();
19 | /**
20 | * 21 | * REQUIRED: session_handle must be returned by a CreateSession call 22 | * to the same master service. 23 | *24 | * 25 | *
string session_handle = 1;
26 | */
27 | com.google.protobuf.ByteString
28 | getSessionHandleBytes();
29 |
30 | /**
31 | * 32 | * Options that define the behavior of the created callable. 33 | *34 | * 35 | *
.tensorflow.CallableOptions options = 2;
36 | */
37 | boolean hasOptions();
38 | /**
39 | * 40 | * Options that define the behavior of the created callable. 41 | *42 | * 43 | *
.tensorflow.CallableOptions options = 2;
44 | */
45 | org.tensorflow.framework.CallableOptions getOptions();
46 | /**
47 | * 48 | * Options that define the behavior of the created callable. 49 | *50 | * 51 | *
.tensorflow.CallableOptions options = 2;
52 | */
53 | org.tensorflow.framework.CallableOptionsOrBuilder getOptionsOrBuilder();
54 | }
55 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/protobuf/replay_log.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | option cc_enable_arenas = true;
4 | package tensorflow;
5 |
6 | import "tensorflow/core/framework/graph.proto";
7 | import "tensorflow/core/protobuf/cluster.proto";
8 | import "tensorflow/core/protobuf/master.proto";
9 |
10 | // Records the creation of a new replay session. We record the device listing
11 | // here to capture the state of the cluster.
12 | message NewReplaySession {
13 | ListDevicesResponse devices = 1;
14 | string session_handle = 2;
15 | }
16 |
17 | message ReplayOp {
18 | double start_time_us = 31;
19 | double end_time_us = 32;
20 |
21 | oneof op {
22 | CreateSessionRequest create_session = 1;
23 | ExtendSessionRequest extend_session = 2;
24 | PartialRunSetupRequest partial_run_setup = 3;
25 | RunStepRequest run_step = 4;
26 | CloseSessionRequest close_session = 5;
27 | ListDevicesRequest list_devices = 6;
28 | ResetRequest reset_request = 7;
29 | MakeCallableRequest make_callable = 8;
30 | RunCallableRequest run_callable = 9;
31 | ReleaseCallableRequest release_callable = 10;
32 | NewReplaySession new_replay_session = 11;
33 | }
34 |
35 | oneof response {
36 | CreateSessionResponse create_session_response = 21;
37 | ExtendSessionResponse extend_session_response = 22;
38 | PartialRunSetupResponse partial_run_setup_response = 23;
39 | RunStepResponse run_step_response = 24;
40 | CloseSessionResponse close_session_response = 25;
41 | ListDevicesResponse list_devices_response = 26;
42 | ResetResponse reset_request_response = 27;
43 | MakeCallableResponse make_callable_response = 28;
44 | RunCallableResponse run_callable_response = 29;
45 | ReleaseCallableResponse release_callable_response = 30;
46 | }
47 | }
48 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/TraceOptsOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface TraceOptsOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.TraceOpts)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Length of the trace to be taken, in seconds. 13 | *14 | * 15 | *
double duration = 1;
16 | */
17 | double getDuration();
18 |
19 | /**
20 | * 21 | * If true, capture step profile locally in each worker. Currently 22 | * unimplemented. 23 | *24 | * 25 | *
bool use_step_profiler = 2;
26 | */
27 | boolean getUseStepProfiler();
28 |
29 | /**
30 | * 31 | * If true, capture kernel events from each worker. 32 | *33 | * 34 | *
bool use_kernel_profiler = 3;
35 | */
36 | boolean getUseKernelProfiler();
37 |
38 | /**
39 | * 40 | * If true, capture extended profiling events from TensorFlow process. 41 | *42 | * 43 | *
bool use_extended_profiler = 4;
44 | */
45 | boolean getUseExtendedProfiler();
46 |
47 | /**
48 | * 49 | * If true, capture GPU profiling events locally on each 50 | * machine. Currently unimplemented. 51 | *52 | * 53 | *
bool use_gpu_profiler = 5;
54 | */
55 | boolean getUseGpuProfiler();
56 |
57 | /**
58 | * 59 | * If true, collect sampled profile events. Currently unimplemented. 60 | *61 | * 62 | *
bool use_sample_profiler = 6;
63 | */
64 | boolean getUseSampleProfiler();
65 | }
66 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/DebuggedSourceFilesOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/debug.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface DebuggedSourceFilesOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.DebuggedSourceFiles)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * A collection of source code files. 13 | *14 | * 15 | *
repeated .tensorflow.DebuggedSourceFile source_files = 1;
16 | */
17 | java.util.List21 | * A collection of source code files. 22 | *23 | * 24 | *
repeated .tensorflow.DebuggedSourceFile source_files = 1;
25 | */
26 | org.tensorflow.framework.DebuggedSourceFile getSourceFiles(int index);
27 | /**
28 | * 29 | * A collection of source code files. 30 | *31 | * 32 | *
repeated .tensorflow.DebuggedSourceFile source_files = 1;
33 | */
34 | int getSourceFilesCount();
35 | /**
36 | * 37 | * A collection of source code files. 38 | *39 | * 40 | *
repeated .tensorflow.DebuggedSourceFile source_files = 1;
41 | */
42 | java.util.List
43 | getSourceFilesOrBuilderList();
44 | /**
45 | * 46 | * A collection of source code files. 47 | *48 | * 49 | *
repeated .tensorflow.DebuggedSourceFile source_files = 1;
50 | */
51 | org.tensorflow.framework.DebuggedSourceFileOrBuilder getSourceFilesOrBuilder(
52 | int index);
53 | }
54 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/remote_fused_graph_execute_info.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "RemoteFusedGraphExecuteInfoProto";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 | import "tensorflow/core/framework/graph.proto";
10 | import "tensorflow/core/framework/tensor_shape.proto";
11 | import "tensorflow/core/framework/types.proto";
12 |
13 | // Protocol buffer representing a handle to a tensorflow resource. Handles are
14 | // not valid across executions, but can be serialized back and forth from within
15 | // a single run.
16 | message RemoteFusedGraphExecuteInfo {
17 |
18 | message TensorShapeTypeProto {
19 | DataType dtype = 1;
20 | TensorShapeProto shape = 2;
21 | }
22 |
23 | // Definition of remote graph
24 | GraphDef remote_graph = 1;
25 |
26 | // Remote fused graph input node name
27 | repeated string graph_input_node_name = 2;
28 |
29 | // Remote fused graph output node name
30 | repeated string graph_output_node_name = 3;
31 |
32 | // Executor's name
33 | string executor_name = 4;
34 |
35 | // Optional: Parameters given to the executor
36 | bytes serialized_executor_parameters = 5;
37 |
38 | // Optional: Default graph input tensor shape used to allocate memory
39 | // before executing op
40 | repeated TensorShapeTypeProto default_graph_input_tensor_shape = 6;
41 |
42 | // Optional: Default graph input tensor shape used to allocate memory
43 | // before executing op
44 | // TODO(satok): Remote output tensor shape once shape information is stored
45 | // in NodeDef
46 | repeated TensorShapeTypeProto default_graph_output_tensor_shape = 7;
47 | };
48 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/AssetFileDefOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/meta_graph.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface AssetFileDefOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.AssetFileDef)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The tensor to bind the asset filename to. 13 | *14 | * 15 | *
.tensorflow.TensorInfo tensor_info = 1;
16 | */
17 | boolean hasTensorInfo();
18 | /**
19 | * 20 | * The tensor to bind the asset filename to. 21 | *22 | * 23 | *
.tensorflow.TensorInfo tensor_info = 1;
24 | */
25 | org.tensorflow.framework.TensorInfo getTensorInfo();
26 | /**
27 | * 28 | * The tensor to bind the asset filename to. 29 | *30 | * 31 | *
.tensorflow.TensorInfo tensor_info = 1;
32 | */
33 | org.tensorflow.framework.TensorInfoOrBuilder getTensorInfoOrBuilder();
34 |
35 | /**
36 | * 37 | * The filename within an assets directory. Note: does not include the path 38 | * prefix, i.e. directories. For an asset at /tmp/path/vocab.txt, the filename 39 | * would be "vocab.txt". 40 | *41 | * 42 | *
string filename = 2;
43 | */
44 | java.lang.String getFilename();
45 | /**
46 | * 47 | * The filename within an assets directory. Note: does not include the path 48 | * prefix, i.e. directories. For an asset at /tmp/path/vocab.txt, the filename 49 | * would be "vocab.txt". 50 | *51 | * 52 | *
string filename = 2;
53 | */
54 | com.google.protobuf.ByteString
55 | getFilenameBytes();
56 | }
57 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/MemoryLogTensorAllocationOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/log_memory.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface MemoryLogTensorAllocationOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.MemoryLogTensorAllocation)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Process-unique step id. 13 | *14 | * 15 | *
int64 step_id = 1;
16 | */
17 | long getStepId();
18 |
19 | /**
20 | * 21 | * Name of the kernel making the allocation as set in GraphDef, 22 | * e.g., "affine2/weights/Assign". 23 | *24 | * 25 | *
string kernel_name = 2;
26 | */
27 | java.lang.String getKernelName();
28 | /**
29 | * 30 | * Name of the kernel making the allocation as set in GraphDef, 31 | * e.g., "affine2/weights/Assign". 32 | *33 | * 34 | *
string kernel_name = 2;
35 | */
36 | com.google.protobuf.ByteString
37 | getKernelNameBytes();
38 |
39 | /**
40 | * 41 | * Allocated tensor details. 42 | *43 | * 44 | *
.tensorflow.TensorDescription tensor = 3;
45 | */
46 | boolean hasTensor();
47 | /**
48 | * 49 | * Allocated tensor details. 50 | *51 | * 52 | *
.tensorflow.TensorDescription tensor = 3;
53 | */
54 | org.tensorflow.framework.TensorDescription getTensor();
55 | /**
56 | * 57 | * Allocated tensor details. 58 | *59 | * 60 | *
.tensorflow.TensorDescription tensor = 3;
61 | */
62 | org.tensorflow.framework.TensorDescriptionOrBuilder getTensorOrBuilder();
63 | }
64 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/FixedLenFeatureProtoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/example_parser_configuration.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface FixedLenFeatureProtoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.FixedLenFeatureProto)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * .tensorflow.DataType dtype = 1;
12 | */
13 | int getDtypeValue();
14 | /**
15 | * .tensorflow.DataType dtype = 1;
16 | */
17 | org.tensorflow.framework.DataType getDtype();
18 |
19 | /**
20 | * .tensorflow.TensorShapeProto shape = 2;
21 | */
22 | boolean hasShape();
23 | /**
24 | * .tensorflow.TensorShapeProto shape = 2;
25 | */
26 | org.tensorflow.framework.TensorShapeProto getShape();
27 | /**
28 | * .tensorflow.TensorShapeProto shape = 2;
29 | */
30 | org.tensorflow.framework.TensorShapeProtoOrBuilder getShapeOrBuilder();
31 |
32 | /**
33 | * .tensorflow.TensorProto default_value = 3;
34 | */
35 | boolean hasDefaultValue();
36 | /**
37 | * .tensorflow.TensorProto default_value = 3;
38 | */
39 | org.tensorflow.framework.TensorProto getDefaultValue();
40 | /**
41 | * .tensorflow.TensorProto default_value = 3;
42 | */
43 | org.tensorflow.framework.TensorProtoOrBuilder getDefaultValueOrBuilder();
44 |
45 | /**
46 | * string values_output_tensor_name = 4;
47 | */
48 | java.lang.String getValuesOutputTensorName();
49 | /**
50 | * string values_output_tensor_name = 4;
51 | */
52 | com.google.protobuf.ByteString
53 | getValuesOutputTensorNameBytes();
54 | }
55 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/AllocationDescriptionOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/allocation_description.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface AllocationDescriptionOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.AllocationDescription)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Total number of bytes requested 13 | *14 | * 15 | *
int64 requested_bytes = 1;
16 | */
17 | long getRequestedBytes();
18 |
19 | /**
20 | * 21 | * Total number of bytes allocated if known 22 | *23 | * 24 | *
int64 allocated_bytes = 2;
25 | */
26 | long getAllocatedBytes();
27 |
28 | /**
29 | * 30 | * Name of the allocator used 31 | *32 | * 33 | *
string allocator_name = 3;
34 | */
35 | java.lang.String getAllocatorName();
36 | /**
37 | * 38 | * Name of the allocator used 39 | *40 | * 41 | *
string allocator_name = 3;
42 | */
43 | com.google.protobuf.ByteString
44 | getAllocatorNameBytes();
45 |
46 | /**
47 | * 48 | * Identifier of the allocated buffer if known 49 | *50 | * 51 | *
int64 allocation_id = 4;
52 | */
53 | long getAllocationId();
54 |
55 | /**
56 | * 57 | * Set if this tensor only has one remaining reference 58 | *59 | * 60 | *
bool has_single_reference = 5;
61 | */
62 | boolean getHasSingleReference();
63 |
64 | /**
65 | * 66 | * Address of the allocation. 67 | *68 | * 69 | *
uint64 ptr = 6;
70 | */
71 | long getPtr();
72 | }
73 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/tensor_shape.proto:
--------------------------------------------------------------------------------
1 | // Protocol buffer representing the shape of tensors.
2 |
3 | syntax = "proto3";
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "TensorShapeProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 |
10 | package tensorflow;
11 |
12 | // Dimensions of a tensor.
13 | message TensorShapeProto {
14 | // One dimension of the tensor.
15 | message Dim {
16 | // Size of the tensor in that dimension.
17 | // This value must be >= -1, but values of -1 are reserved for "unknown"
18 | // shapes (values of -1 mean "unknown" dimension). Certain wrappers
19 | // that work with TensorShapeProto may fail at runtime when deserializing
20 | // a TensorShapeProto containing a dim value of -1.
21 | int64 size = 1;
22 |
23 | // Optional name of the tensor dimension.
24 | string name = 2;
25 | };
26 |
27 | // Dimensions of the tensor, such as {"input", 30}, {"output", 40}
28 | // for a 30 x 40 2D tensor. If an entry has size -1, this
29 | // corresponds to a dimension of unknown size. The names are
30 | // optional.
31 | //
32 | // The order of entries in "dim" matters: It indicates the layout of the
33 | // values in the tensor in-memory representation.
34 | //
35 | // The first entry in "dim" is the outermost dimension used to layout the
36 | // values, the last entry is the innermost dimension. This matches the
37 | // in-memory layout of RowMajor Eigen tensors.
38 | //
39 | // If "dim.size()" > 0, "unknown_rank" must be false.
40 | repeated Dim dim = 2;
41 |
42 | // If true, the number of dimensions in the shape is unknown.
43 | //
44 | // If true, "dim.size()" must be 0.
45 | bool unknown_rank = 3;
46 | };
47 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/RecvBufResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface RecvBufResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.RecvBufResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Address of source field on server. 13 | *14 | * 15 | *
fixed64 buf_ptr = 1;
16 | */
17 | long getBufPtr();
18 |
19 | /**
20 | * 21 | * Byte length of buf_ptr field, if set. 22 | *23 | * 24 | *
int64 num_bytes = 2;
25 | */
26 | long getNumBytes();
27 |
28 | /**
29 | * 30 | * True if value is 'dead' like a tensor. 31 | *32 | * 33 | *
bool is_dead = 3;
34 | */
35 | boolean getIsDead();
36 |
37 | /**
38 | * 39 | * Optional, implementation-specific data. 40 | *41 | * 42 | *
.google.protobuf.Any transport_options = 4;
43 | */
44 | boolean hasTransportOptions();
45 | /**
46 | * 47 | * Optional, implementation-specific data. 48 | *49 | * 50 | *
.google.protobuf.Any transport_options = 4;
51 | */
52 | com.google.protobuf.Any getTransportOptions();
53 | /**
54 | * 55 | * Optional, implementation-specific data. 56 | *57 | * 58 | *
.google.protobuf.Any transport_options = 4;
59 | */
60 | com.google.protobuf.AnyOrBuilder getTransportOptionsOrBuilder();
61 |
62 | /**
63 | * 64 | * Optional, for timeline. 65 | *66 | * 67 | *
int64 send_start_micros = 5;
68 | */
69 | long getSendStartMicros();
70 | }
71 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/DeregisterGraphRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface DeregisterGraphRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.DeregisterGraphRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The session_handle used when registering the graph. If session_handle is 13 | * empty, a single global namespace is used. 14 | *15 | * 16 | *
string session_handle = 2;
17 | */
18 | java.lang.String getSessionHandle();
19 | /**
20 | * 21 | * The session_handle used when registering the graph. If session_handle is 22 | * empty, a single global namespace is used. 23 | *24 | * 25 | *
string session_handle = 2;
26 | */
27 | com.google.protobuf.ByteString
28 | getSessionHandleBytes();
29 |
30 | /**
31 | * 32 | * Set to true if `CreateWorkerSession` was called for `session_handle`. 33 | *34 | * 35 | *
bool create_worker_session_called = 3;
36 | */
37 | boolean getCreateWorkerSessionCalled();
38 |
39 | /**
40 | * 41 | * REQUIRED: graph_handle must be returned by a RegisterGraph call 42 | * to the same WorkerService. 43 | *44 | * 45 | *
string graph_handle = 1;
46 | */
47 | java.lang.String getGraphHandle();
48 | /**
49 | * 50 | * REQUIRED: graph_handle must be returned by a RegisterGraph call 51 | * to the same WorkerService. 52 | *53 | * 54 | *
string graph_handle = 1;
55 | */
56 | com.google.protobuf.ByteString
57 | getGraphHandleBytes();
58 | }
59 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/CreateWorkerSessionRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface CreateWorkerSessionRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CreateWorkerSessionRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Sessions are identified by a given handle. 13 | *14 | * 15 | *
string session_handle = 1;
16 | */
17 | java.lang.String getSessionHandle();
18 | /**
19 | * 20 | * Sessions are identified by a given handle. 21 | *22 | * 23 | *
string session_handle = 1;
24 | */
25 | com.google.protobuf.ByteString
26 | getSessionHandleBytes();
27 |
28 | /**
29 | * 30 | * Defines the configuration of a TensorFlow worker. 31 | *32 | * 33 | *
.tensorflow.ServerDef server_def = 2;
34 | */
35 | boolean hasServerDef();
36 | /**
37 | * 38 | * Defines the configuration of a TensorFlow worker. 39 | *40 | * 41 | *
.tensorflow.ServerDef server_def = 2;
42 | */
43 | org.tensorflow.distruntime.ServerDef getServerDef();
44 | /**
45 | * 46 | * Defines the configuration of a TensorFlow worker. 47 | *48 | * 49 | *
.tensorflow.ServerDef server_def = 2;
50 | */
51 | org.tensorflow.distruntime.ServerDefOrBuilder getServerDefOrBuilder();
52 |
53 | /**
54 | * 55 | * If true, any resources such as Variables used in the session will not be 56 | * shared with other sessions. 57 | *58 | * 59 | *
bool isolate_session_state = 3;
60 | */
61 | boolean getIsolateSessionState();
62 | }
63 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/FeaturesOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/feature.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface FeaturesOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.Features)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Map from feature name to feature. 13 | *14 | * 15 | *
map<string, .tensorflow.Feature> feature = 1;
16 | */
17 | int getFeatureCount();
18 | /**
19 | * 20 | * Map from feature name to feature. 21 | *22 | * 23 | *
map<string, .tensorflow.Feature> feature = 1;
24 | */
25 | boolean containsFeature(
26 | java.lang.String key);
27 | /**
28 | * Use {@link #getFeatureMap()} instead.
29 | */
30 | @java.lang.Deprecated
31 | java.util.Map35 | * Map from feature name to feature. 36 | *37 | * 38 | *
map<string, .tensorflow.Feature> feature = 1;
39 | */
40 | java.util.Map44 | * Map from feature name to feature. 45 | *46 | * 47 | *
map<string, .tensorflow.Feature> feature = 1;
48 | */
49 |
50 | org.tensorflow.example.Feature getFeatureOrDefault(
51 | java.lang.String key,
52 | org.tensorflow.example.Feature defaultValue);
53 | /**
54 | * 55 | * Map from feature name to feature. 56 | *57 | * 58 | *
map<string, .tensorflow.Feature> feature = 1;
59 | */
60 |
61 | org.tensorflow.example.Feature getFeatureOrThrow(
62 | java.lang.String key);
63 | }
64 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/protobuf/saver.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "SaverProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.util";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
9 |
10 | // Protocol buffer representing the configuration of a Saver.
11 | message SaverDef {
12 | // The name of the tensor in which to specify the filename when saving or
13 | // restoring a model checkpoint.
14 | string filename_tensor_name = 1;
15 |
16 | // The operation to run when saving a model checkpoint.
17 | string save_tensor_name = 2;
18 |
19 | // The operation to run when restoring a model checkpoint.
20 | string restore_op_name = 3;
21 |
22 | // Maximum number of checkpoints to keep. If 0, no checkpoints are deleted.
23 | int32 max_to_keep = 4;
24 |
25 | // Shard the save files, one per device that has Variable nodes.
26 | bool sharded = 5;
27 |
28 | // How often to keep an additional checkpoint. If not specified, only the last
29 | // "max_to_keep" checkpoints are kept; if specified, in addition to keeping
30 | // the last "max_to_keep" checkpoints, an additional checkpoint will be kept
31 | // for every n hours of training.
32 | float keep_checkpoint_every_n_hours = 6;
33 |
34 | // A version number that identifies a different on-disk checkpoint format.
35 | // Usually, each subclass of BaseSaverBuilder works with a particular
36 | // version/format. However, it is possible that the same builder may be
37 | // upgraded to support a newer checkpoint format in the future.
38 | enum CheckpointFormatVersion {
39 | // Internal legacy format.
40 | LEGACY = 0;
41 | // Deprecated format: tf.Saver() which works with tensorflow::table::Table.
42 | V1 = 1;
43 | // Current format: more efficient.
44 | V2 = 2;
45 | }
46 | CheckpointFormatVersion version = 7;
47 | }
48 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/MemoryLogTensorOutputOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/log_memory.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface MemoryLogTensorOutputOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.MemoryLogTensorOutput)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Process-unique step id. 13 | *14 | * 15 | *
int64 step_id = 1;
16 | */
17 | long getStepId();
18 |
19 | /**
20 | * 21 | * Name of the kernel producing an output as set in GraphDef, e.g., 22 | * "affine2/weights/Assign". 23 | *24 | * 25 | *
string kernel_name = 2;
26 | */
27 | java.lang.String getKernelName();
28 | /**
29 | * 30 | * Name of the kernel producing an output as set in GraphDef, e.g., 31 | * "affine2/weights/Assign". 32 | *33 | * 34 | *
string kernel_name = 2;
35 | */
36 | com.google.protobuf.ByteString
37 | getKernelNameBytes();
38 |
39 | /**
40 | * 41 | * Index of the output being set. 42 | *43 | * 44 | *
int32 index = 3;
45 | */
46 | int getIndex();
47 |
48 | /**
49 | * 50 | * Output tensor details. 51 | *52 | * 53 | *
.tensorflow.TensorDescription tensor = 4;
54 | */
55 | boolean hasTensor();
56 | /**
57 | * 58 | * Output tensor details. 59 | *60 | * 61 | *
.tensorflow.TensorDescription tensor = 4;
62 | */
63 | org.tensorflow.framework.TensorDescription getTensor();
64 | /**
65 | * 66 | * Output tensor details. 67 | *68 | * 69 | *
.tensorflow.TensorDescription tensor = 4;
70 | */
71 | org.tensorflow.framework.TensorDescriptionOrBuilder getTensorOrBuilder();
72 | }
73 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/session_service.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | option cc_enable_arenas = true;
4 |
5 | import "tensorflow_serving/apis/model.proto";
6 | import "tensorflow/core/protobuf/config.proto";
7 | import "tensorflow/core/protobuf/named_tensor.proto";
8 |
9 | package tensorflow.serving;
10 |
11 | message SessionRunRequest {
12 | // Model Specification. If version is not specified, will use the latest
13 | // (numerical) version.
14 | ModelSpec model_spec = 1;
15 |
16 | // Tensors to be fed in the step. Each feed is a named tensor.
17 | repeated NamedTensorProto feed = 2;
18 |
19 | // Fetches. A list of tensor names. The caller expects a tensor to
20 | // be returned for each fetch[i] (see RunResponse.tensor). The
21 | // order of specified fetches does not change the execution order.
22 | repeated string fetch = 3;
23 |
24 | // Target Nodes. A list of node names. The named nodes will be run
25 | // to but their outputs will not be fetched.
26 | repeated string target = 4;
27 |
28 | // Options for the run call. **Currently ignored.**
29 | RunOptions options = 5;
30 | }
31 |
32 | message SessionRunResponse {
33 | // Effective Model Specification used for session run.
34 | ModelSpec model_spec = 3;
35 |
36 | // NOTE: The order of the returned tensors may or may not match
37 | // the fetch order specified in RunRequest.
38 | repeated NamedTensorProto tensor = 1;
39 |
40 | // Returned metadata if requested in the options.
41 | RunMetadata metadata = 2;
42 | }
43 |
44 | // SessionService defines a service with which a client can interact to execute
45 | // Tensorflow model inference. The SessionService::SessionRun method is similar
46 | // to MasterService::RunStep of Tensorflow, except that all sessions are ready
47 | // to run, and you request a specific model/session with ModelSpec.
48 | service SessionService {
49 | // Runs inference of a given model.
50 | rpc SessionRun(SessionRunRequest) returns (SessionRunResponse);
51 | }
52 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/MemoryLogRawDeallocationOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/log_memory.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface MemoryLogRawDeallocationOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.MemoryLogRawDeallocation)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Process-unique step id. 13 | *14 | * 15 | *
int64 step_id = 1;
16 | */
17 | long getStepId();
18 |
19 | /**
20 | * 21 | * Name of the operation making the deallocation. 22 | *23 | * 24 | *
string operation = 2;
25 | */
26 | java.lang.String getOperation();
27 | /**
28 | * 29 | * Name of the operation making the deallocation. 30 | *31 | * 32 | *
string operation = 2;
33 | */
34 | com.google.protobuf.ByteString
35 | getOperationBytes();
36 |
37 | /**
38 | * 39 | * Id of the tensor buffer being deallocated, used to match to a 40 | * corresponding allocation. 41 | *42 | * 43 | *
int64 allocation_id = 3;
44 | */
45 | long getAllocationId();
46 |
47 | /**
48 | * 49 | * Name of the allocator used. 50 | *51 | * 52 | *
string allocator_name = 4;
53 | */
54 | java.lang.String getAllocatorName();
55 | /**
56 | * 57 | * Name of the allocator used. 58 | *59 | * 60 | *
string allocator_name = 4;
61 | */
62 | com.google.protobuf.ByteString
63 | getAllocatorNameBytes();
64 |
65 | /**
66 | * 67 | * True if the deallocation is queued and will be performed later, 68 | * e.g. for GPU lazy freeing of buffers. 69 | *70 | * 71 | *
bool deferred = 5;
72 | */
73 | boolean getDeferred();
74 | }
75 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/LoggingRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface LoggingRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.LoggingRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * If true, RPC logging will be enabled. 13 | *14 | * 15 | *
bool enable_rpc_logging = 1;
16 | */
17 | boolean getEnableRpcLogging();
18 |
19 | /**
20 | * 21 | * If true, RPC logging will be disabled. 22 | *23 | * 24 | *
bool disable_rpc_logging = 4;
25 | */
26 | boolean getDisableRpcLogging();
27 |
28 | /**
29 | * 30 | * If true, discard any saved logging data (for all steps). 31 | *32 | * 33 | *
bool clear = 2;
34 | */
35 | boolean getClear();
36 |
37 | /**
38 | * 39 | * When set, requests all saved log data pertaining to the step. 40 | * Any log data retrieved is eliminated from the store and cannot be 41 | * retrieved again. 42 | *43 | * 44 | *
repeated int64 fetch_step_id = 3;
45 | */
46 | java.util.List49 | * When set, requests all saved log data pertaining to the step. 50 | * Any log data retrieved is eliminated from the store and cannot be 51 | * retrieved again. 52 | *53 | * 54 | *
repeated int64 fetch_step_id = 3;
55 | */
56 | int getFetchStepIdCount();
57 | /**
58 | * 59 | * When set, requests all saved log data pertaining to the step. 60 | * Any log data retrieved is eliminated from the store and cannot be 61 | * retrieved again. 62 | *63 | * 64 | *
repeated int64 fetch_step_id = 3;
65 | */
66 | long getFetchStepId(int index);
67 | }
68 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/MemoryStatsOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/step_stats.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface MemoryStatsOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.MemoryStats)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int64 temp_memory_size = 1;
12 | */
13 | long getTempMemorySize();
14 |
15 | /**
16 | * int64 persistent_memory_size = 3;
17 | */
18 | long getPersistentMemorySize();
19 |
20 | /**
21 | * repeated int64 persistent_tensor_alloc_ids = 5;
22 | */
23 | java.util.Listrepeated int64 persistent_tensor_alloc_ids = 5;
26 | */
27 | int getPersistentTensorAllocIdsCount();
28 | /**
29 | * repeated int64 persistent_tensor_alloc_ids = 5;
30 | */
31 | long getPersistentTensorAllocIds(int index);
32 |
33 | /**
34 | * int64 device_temp_memory_size = 2 [deprecated = true];
35 | */
36 | @java.lang.Deprecated long getDeviceTempMemorySize();
37 |
38 | /**
39 | * int64 device_persistent_memory_size = 4 [deprecated = true];
40 | */
41 | @java.lang.Deprecated long getDevicePersistentMemorySize();
42 |
43 | /**
44 | * repeated int64 device_persistent_tensor_alloc_ids = 6 [deprecated = true];
45 | */
46 | @java.lang.Deprecated java.util.Listrepeated int64 device_persistent_tensor_alloc_ids = 6 [deprecated = true];
49 | */
50 | @java.lang.Deprecated int getDevicePersistentTensorAllocIdsCount();
51 | /**
52 | * repeated int64 device_persistent_tensor_alloc_ids = 6 [deprecated = true];
53 | */
54 | @java.lang.Deprecated long getDevicePersistentTensorAllocIds(int index);
55 | }
56 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/SavedModelOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/saved_model.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface SavedModelOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.SavedModel)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The schema version of the SavedModel instance. Used for versioning when 13 | * making future changes to the specification/implementation. Initial value 14 | * at release will be 1. 15 | *16 | * 17 | *
int64 saved_model_schema_version = 1;
18 | */
19 | long getSavedModelSchemaVersion();
20 |
21 | /**
22 | * 23 | * One or more MetaGraphs. 24 | *25 | * 26 | *
repeated .tensorflow.MetaGraphDef meta_graphs = 2;
27 | */
28 | java.util.List32 | * One or more MetaGraphs. 33 | *34 | * 35 | *
repeated .tensorflow.MetaGraphDef meta_graphs = 2;
36 | */
37 | org.tensorflow.framework.MetaGraphDef getMetaGraphs(int index);
38 | /**
39 | * 40 | * One or more MetaGraphs. 41 | *42 | * 43 | *
repeated .tensorflow.MetaGraphDef meta_graphs = 2;
44 | */
45 | int getMetaGraphsCount();
46 | /**
47 | * 48 | * One or more MetaGraphs. 49 | *50 | * 51 | *
repeated .tensorflow.MetaGraphDef meta_graphs = 2;
52 | */
53 | java.util.List
54 | getMetaGraphsOrBuilderList();
55 | /**
56 | * 57 | * One or more MetaGraphs. 58 | *59 | * 60 | *
repeated .tensorflow.MetaGraphDef meta_graphs = 2;
61 | */
62 | org.tensorflow.framework.MetaGraphDefOrBuilder getMetaGraphsOrBuilder(
63 | int index);
64 | }
65 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/example/FeatureListsOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/example/feature.proto
3 |
4 | package org.tensorflow.example;
5 |
6 | public interface FeatureListsOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.FeatureLists)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Map from feature name to feature list. 13 | *14 | * 15 | *
map<string, .tensorflow.FeatureList> feature_list = 1;
16 | */
17 | int getFeatureListCount();
18 | /**
19 | * 20 | * Map from feature name to feature list. 21 | *22 | * 23 | *
map<string, .tensorflow.FeatureList> feature_list = 1;
24 | */
25 | boolean containsFeatureList(
26 | java.lang.String key);
27 | /**
28 | * Use {@link #getFeatureListMap()} instead.
29 | */
30 | @java.lang.Deprecated
31 | java.util.Map35 | * Map from feature name to feature list. 36 | *37 | * 38 | *
map<string, .tensorflow.FeatureList> feature_list = 1;
39 | */
40 | java.util.Map44 | * Map from feature name to feature list. 45 | *46 | * 47 | *
map<string, .tensorflow.FeatureList> feature_list = 1;
48 | */
49 |
50 | org.tensorflow.example.FeatureList getFeatureListOrDefault(
51 | java.lang.String key,
52 | org.tensorflow.example.FeatureList defaultValue);
53 | /**
54 | * 55 | * Map from feature name to feature list. 56 | *57 | * 58 | *
map<string, .tensorflow.FeatureList> feature_list = 1;
59 | */
60 |
61 | org.tensorflow.example.FeatureList getFeatureListOrThrow(
62 | java.lang.String key);
63 | }
64 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/MemoryLogRawAllocationOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/log_memory.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface MemoryLogRawAllocationOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.MemoryLogRawAllocation)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Process-unique step id. 13 | *14 | * 15 | *
int64 step_id = 1;
16 | */
17 | long getStepId();
18 |
19 | /**
20 | * 21 | * Name of the operation making the allocation. 22 | *23 | * 24 | *
string operation = 2;
25 | */
26 | java.lang.String getOperation();
27 | /**
28 | * 29 | * Name of the operation making the allocation. 30 | *31 | * 32 | *
string operation = 2;
33 | */
34 | com.google.protobuf.ByteString
35 | getOperationBytes();
36 |
37 | /**
38 | * 39 | * Number of bytes in the allocation. 40 | *41 | * 42 | *
int64 num_bytes = 3;
43 | */
44 | long getNumBytes();
45 |
46 | /**
47 | * 48 | * Address of the allocation. 49 | *50 | * 51 | *
uint64 ptr = 4;
52 | */
53 | long getPtr();
54 |
55 | /**
56 | * 57 | * Id of the tensor buffer being allocated, used to match to a 58 | * corresponding deallocation. 59 | *60 | * 61 | *
int64 allocation_id = 5;
62 | */
63 | long getAllocationId();
64 |
65 | /**
66 | * 67 | * Name of the allocator used. 68 | *69 | * 70 | *
string allocator_name = 6;
71 | */
72 | java.lang.String getAllocatorName();
73 | /**
74 | * 75 | * Name of the allocator used. 76 | *77 | * 78 | *
string allocator_name = 6;
79 | */
80 | com.google.protobuf.ByteString
81 | getAllocatorNameBytes();
82 | }
83 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/CleanupAllRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface CleanupAllRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CleanupAllRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * A list of container names. 13 | * If 'container' is not empty, releases resources in the given 14 | * containers in all devices. 15 | * If 'container' is empty, releases resources in the default 16 | * container in all devices. 17 | *18 | * 19 | *
repeated string container = 1;
20 | */
21 | java.util.List25 | * A list of container names. 26 | * If 'container' is not empty, releases resources in the given 27 | * containers in all devices. 28 | * If 'container' is empty, releases resources in the default 29 | * container in all devices. 30 | *31 | * 32 | *
repeated string container = 1;
33 | */
34 | int getContainerCount();
35 | /**
36 | * 37 | * A list of container names. 38 | * If 'container' is not empty, releases resources in the given 39 | * containers in all devices. 40 | * If 'container' is empty, releases resources in the default 41 | * container in all devices. 42 | *43 | * 44 | *
repeated string container = 1;
45 | */
46 | java.lang.String getContainer(int index);
47 | /**
48 | * 49 | * A list of container names. 50 | * If 'container' is not empty, releases resources in the given 51 | * containers in all devices. 52 | * If 'container' is empty, releases resources in the default 53 | * container in all devices. 54 | *55 | * 56 | *
repeated string container = 1;
57 | */
58 | com.google.protobuf.ByteString
59 | getContainerBytes(int index);
60 | }
61 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/protobuf/tensorflow_server.proto:
--------------------------------------------------------------------------------
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 | syntax = "proto3";
17 |
18 | import "tensorflow/core/protobuf/config.proto";
19 | import "tensorflow/core/protobuf/cluster.proto";
20 |
21 | package tensorflow;
22 | option cc_enable_arenas = true;
23 | option java_outer_classname = "ServerProtos";
24 | option java_multiple_files = true;
25 | option java_package = "org.tensorflow.distruntime";
26 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
27 | // Defines the configuration of a single TensorFlow server.
28 | message ServerDef {
29 | // The cluster of which this server is a member.
30 | ClusterDef cluster = 1;
31 |
32 | // The name of the job of which this server is a member.
33 | //
34 | // NOTE(mrry): The `cluster` field must contain a `JobDef` with a `name` field
35 | // that matches this name.
36 | string job_name = 2;
37 |
38 | // The task index of this server in its job.
39 | //
40 | // NOTE: The `cluster` field must contain a `JobDef` with a matching `name`
41 | // and a mapping in its `tasks` field for this index.
42 | int32 task_index = 3;
43 |
44 | // The default configuration for sessions that run on this server.
45 | ConfigProto default_session_config = 4;
46 |
47 | // The protocol to be used by this server.
48 | //
49 | // Acceptable values include: "grpc", "grpc+verbs".
50 | string protocol = 5;
51 | }
52 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/resources/resources.proto:
--------------------------------------------------------------------------------
1 | // Representations for resources used by servables, and available in a system.
2 | //
3 | // Each of the string-typed values are free-form, so that they can be extended
4 | // by third parties. However we strongly recommend using the values defined in
5 | // resource_values.h when possible, for standardization.
6 |
7 | syntax = "proto3";
8 |
9 | import "google/protobuf/wrappers.proto";
10 |
11 | package tensorflow.serving;
12 |
13 | // One kind of resource on one device (or type of device).
14 | message Resource {
15 | // The type of device on which the resource resides, e.g. CPU or GPU.
16 | string device = 1;
17 |
18 | // A specific instance of the device of type 'device' to which the resources
19 | // are bound (instances are assumed to be numbered 0, 1, ...).
20 | //
21 | // When representing the resources required by a servable that has yet to be
22 | // loaded, this field is optional. If not set, it denotes that the servable's
23 | // resources are not (yet) bound to a specific instance.
24 | google.protobuf.UInt32Value device_instance = 2;
25 |
26 | // The kind of resource on the device (instance), e.g. RAM or compute share.
27 | //
28 | // A given type of resource should have a standard unit that represents the
29 | // smallest useful quantization. We strongly recommend including the unit
30 | // (e.g. bytes or millicores) in this string, as in "ram_bytes".
31 | string kind = 3;
32 | }
33 |
34 | // An allocation of one or more kinds of resources, along with the quantity of
35 | // each. Used to denote the resources that a servable (or collection of
36 | // servables) will use or is currently using. Also used to denote resources
37 | // available to the serving system for loading more servables.
38 | message ResourceAllocation {
39 | // A collection of resources, each with a quantity. Treated as a resource->
40 | // quantity map, i.e. no resource can repeat and the order is immaterial.
41 | message Entry {
42 | Resource resource = 1;
43 | uint64 quantity = 2;
44 | }
45 | repeated Entry resource_quantities = 1;
46 | }
47 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/protobuf/device_properties.proto:
--------------------------------------------------------------------------------
1 | /* Copyright 2017 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 | syntax = "proto3";
17 |
18 | package tensorflow;
19 | option cc_enable_arenas = true;
20 | option java_outer_classname = "DevicePropertiesProtos";
21 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
22 |
23 | message DeviceProperties {
24 | // Device type (CPU, GPU, ...)
25 | string type = 1;
26 | // Vendor (Intel, nvidia, ...)
27 | string vendor = 2;
28 | // Model (Haswell, K40, ...)
29 | string model = 3;
30 | // Core Frequency in Mhz
31 | int64 frequency = 4;
32 | // Number of cores
33 | int64 num_cores = 5;
34 | // Version of the tools and libraries used with this device (e.g. gcc 4.9,
35 | // cudnn 5.1)
36 | maprepeated .tensorflow.FunctionDef function = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.FunctionDef function = 1;
17 | */
18 | org.tensorflow.framework.FunctionDef getFunction(int index);
19 | /**
20 | * repeated .tensorflow.FunctionDef function = 1;
21 | */
22 | int getFunctionCount();
23 | /**
24 | * repeated .tensorflow.FunctionDef function = 1;
25 | */
26 | java.util.List
27 | getFunctionOrBuilderList();
28 | /**
29 | * repeated .tensorflow.FunctionDef function = 1;
30 | */
31 | org.tensorflow.framework.FunctionDefOrBuilder getFunctionOrBuilder(
32 | int index);
33 |
34 | /**
35 | * repeated .tensorflow.GradientDef gradient = 2;
36 | */
37 | java.util.Listrepeated .tensorflow.GradientDef gradient = 2;
41 | */
42 | org.tensorflow.framework.GradientDef getGradient(int index);
43 | /**
44 | * repeated .tensorflow.GradientDef gradient = 2;
45 | */
46 | int getGradientCount();
47 | /**
48 | * repeated .tensorflow.GradientDef gradient = 2;
49 | */
50 | java.util.List
51 | getGradientOrBuilderList();
52 | /**
53 | * repeated .tensorflow.GradientDef gradient = 2;
54 | */
55 | org.tensorflow.framework.GradientDefOrBuilder getGradientOrBuilder(
56 | int index);
57 | }
58 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/CreateSessionRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface CreateSessionRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CreateSessionRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The initial graph definition. 13 | *14 | * 15 | *
.tensorflow.GraphDef graph_def = 1;
16 | */
17 | boolean hasGraphDef();
18 | /**
19 | * 20 | * The initial graph definition. 21 | *22 | * 23 | *
.tensorflow.GraphDef graph_def = 1;
24 | */
25 | org.tensorflow.framework.GraphDef getGraphDef();
26 | /**
27 | * 28 | * The initial graph definition. 29 | *30 | * 31 | *
.tensorflow.GraphDef graph_def = 1;
32 | */
33 | org.tensorflow.framework.GraphDefOrBuilder getGraphDefOrBuilder();
34 |
35 | /**
36 | * 37 | * Configuration options. 38 | *39 | * 40 | *
.tensorflow.ConfigProto config = 2;
41 | */
42 | boolean hasConfig();
43 | /**
44 | * 45 | * Configuration options. 46 | *47 | * 48 | *
.tensorflow.ConfigProto config = 2;
49 | */
50 | org.tensorflow.framework.ConfigProto getConfig();
51 | /**
52 | * 53 | * Configuration options. 54 | *55 | * 56 | *
.tensorflow.ConfigProto config = 2;
57 | */
58 | org.tensorflow.framework.ConfigProtoOrBuilder getConfigOrBuilder();
59 |
60 | /**
61 | * 62 | * The target string used from the client's perspective. 63 | *64 | * 65 | *
string target = 3;
66 | */
67 | java.lang.String getTarget();
68 | /**
69 | * 70 | * The target string used from the client's perspective. 71 | *72 | * 73 | *
string target = 3;
74 | */
75 | com.google.protobuf.ByteString
76 | getTargetBytes();
77 | }
78 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/SummaryMetadataOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/summary.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface SummaryMetadataOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.SummaryMetadata)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Data that associates a summary with a certain plugin. 13 | *14 | * 15 | *
.tensorflow.SummaryMetadata.PluginData plugin_data = 1;
16 | */
17 | boolean hasPluginData();
18 | /**
19 | * 20 | * Data that associates a summary with a certain plugin. 21 | *22 | * 23 | *
.tensorflow.SummaryMetadata.PluginData plugin_data = 1;
24 | */
25 | org.tensorflow.framework.SummaryMetadata.PluginData getPluginData();
26 | /**
27 | * 28 | * Data that associates a summary with a certain plugin. 29 | *30 | * 31 | *
.tensorflow.SummaryMetadata.PluginData plugin_data = 1;
32 | */
33 | org.tensorflow.framework.SummaryMetadata.PluginDataOrBuilder getPluginDataOrBuilder();
34 |
35 | /**
36 | * 37 | * Display name for viewing in TensorBoard. 38 | *39 | * 40 | *
string display_name = 2;
41 | */
42 | java.lang.String getDisplayName();
43 | /**
44 | * 45 | * Display name for viewing in TensorBoard. 46 | *47 | * 48 | *
string display_name = 2;
49 | */
50 | com.google.protobuf.ByteString
51 | getDisplayNameBytes();
52 |
53 | /**
54 | * 55 | * Longform readable description of the summary sequence. Markdown supported. 56 | *57 | * 58 | *
string summary_description = 3;
59 | */
60 | java.lang.String getSummaryDescription();
61 | /**
62 | * 63 | * Longform readable description of the summary sequence. Markdown supported. 64 | *65 | * 66 | *
string summary_description = 3;
67 | */
68 | com.google.protobuf.ByteString
69 | getSummaryDescriptionBytes();
70 | }
71 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/protobuf/checkpointable_object_graph.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | option cc_enable_arenas = true;
4 |
5 | package tensorflow;
6 |
7 | // A TensorBundle addition which saves extra information about the objects which
8 | // own variables, allowing for more robust checkpoint loading into modified
9 | // programs.
10 |
11 | message CheckpointableObjectGraph {
12 | message CheckpointableObject {
13 | message ObjectReference {
14 | // An index into `CheckpointableObjectGraph.nodes`, indicating the object
15 | // being referenced.
16 | int32 node_id = 1;
17 | // A user-provided name for the edge.
18 | string local_name = 2;
19 | }
20 |
21 | message SerializedTensor {
22 | // A name for the Tensor. Simple variables have only one
23 | // `SerializedTensor` named "VARIABLE_VALUE" by convention. This value may
24 | // be restored on object creation as an optimization.
25 | string name = 1;
26 | // The full name of the variable/tensor, if applicable. Used to allow
27 | // name-based loading of checkpoints which were saved using an
28 | // object-based API. Should match the checkpoint key which would have been
29 | // assigned by tf.train.Saver.
30 | string full_name = 2;
31 | // The generated name of the Tensor in the checkpoint.
32 | string checkpoint_key = 3;
33 | }
34 |
35 | message SlotVariableReference {
36 | // An index into `CheckpointableObjectGraph.nodes`, indicating the
37 | // variable object this slot was created for.
38 | int32 original_variable_node_id = 1;
39 | // The name of the slot (e.g. "m"/"v").
40 | string slot_name = 2;
41 | // An index into `CheckpointableObjectGraph.nodes`, indicating the
42 | // `Object` with the value of the slot variable.
43 | int32 slot_variable_node_id = 3;
44 | }
45 |
46 | // Objects which this object depends on.
47 | repeated ObjectReference children = 1;
48 | // Serialized data specific to this object.
49 | repeated SerializedTensor attributes = 2;
50 | // Slot variables owned by this object.
51 | repeated SlotVariableReference slot_variables = 3;
52 | }
53 |
54 | repeated CheckpointableObject nodes = 1;
55 | }
56 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/input.proto:
--------------------------------------------------------------------------------
1 | // Input used in serving APIs. Based on the tensorflow.Example family of
2 | // feature representations.
3 |
4 | syntax = "proto3";
5 |
6 | option cc_enable_arenas = true;
7 |
8 | import "tensorflow/core/example/example.proto";
9 |
10 | package tensorflow.serving;
11 |
12 | // Specifies one or more fully independent input Examples.
13 | // See examples at:
14 | // https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/example/example.proto
15 | message ExampleList {
16 | repeated tensorflow.Example examples = 1;
17 | }
18 |
19 | // Specifies one or more independent input Examples, with a common context
20 | // Example.
21 | //
22 | // The common use case for context is to cleanly and optimally specify some
23 | // features that are common across multiple examples.
24 | //
25 | // See example below with a search query as the context and multiple restaurants
26 | // to perform some inference on.
27 | //
28 | // context: {
29 | // feature: {
30 | // key : "query"
31 | // value: {
32 | // bytes_list: {
33 | // value: [ "pizza" ]
34 | // }
35 | // }
36 | // }
37 | // }
38 | // examples: {
39 | // feature: {
40 | // key : "cuisine"
41 | // value: {
42 | // bytes_list: {
43 | // value: [ "Pizzeria" ]
44 | // }
45 | // }
46 | // }
47 | // }
48 | // examples: {
49 | // feature: {
50 | // key : "cuisine"
51 | // value: {
52 | // bytes_list: {
53 | // value: [ "Taqueria" ]
54 | // }
55 | // }
56 | // }
57 | // }
58 | //
59 | // Implementations of ExampleListWithContext merge the context Example into each
60 | // of the Examples. Note that feature keys must not be duplicated between the
61 | // Examples and context Example, or the behavior is undefined.
62 | //
63 | // See also:
64 | // tensorflow/core/example/example.proto
65 | // https://developers.google.com/protocol-buffers/docs/proto3#maps
66 | message ExampleListWithContext {
67 | repeated tensorflow.Example examples = 1;
68 | tensorflow.Example context = 2;
69 | }
70 |
71 | message Input {
72 | oneof kind {
73 | ExampleList example_list = 1 [lazy = true];
74 | ExampleListWithContext example_list_with_context = 2 [lazy = true];
75 | }
76 | }
77 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow_serving/apis/inference.proto:
--------------------------------------------------------------------------------
1 | // This file contains messages for various machine learning inferences
2 | // such as regression and classification.
3 | //
4 | // In many applications more than one type of inference is desired for a single
5 | // input. For example, given meteorologic data an application may want to
6 | // perform a classification to determine if we should expect rain, snow or sun
7 | // and also perform a regression to predict the temperature.
8 | // Sharing the single input data between two inference tasks can be accomplished
9 | // using MultiInferenceRequest and MultiInferenceResponse.
10 |
11 | syntax = "proto3";
12 |
13 | option cc_enable_arenas = true;
14 |
15 | import "tensorflow_serving/apis/classification.proto";
16 | import "tensorflow_serving/apis/input.proto";
17 | import "tensorflow_serving/apis/model.proto";
18 | import "tensorflow_serving/apis/regression.proto";
19 |
20 | package tensorflow.serving;
21 |
22 | // Inference request such as classification, regression, etc...
23 | message InferenceTask {
24 | // Model Specification. If version is not specified, will use the latest
25 | // (numerical) version.
26 | // All ModelSpecs in a MultiInferenceRequest must access the same model name.
27 | ModelSpec model_spec = 1;
28 |
29 | // Signature's method_name. Should be one of the method names defined in
30 | // third_party/tensorflow/python/saved_model/signature_constants.py.
31 | // e.g. "tensorflow/serving/classify".
32 | string method_name = 2;
33 | }
34 |
35 | // Inference result, matches the type of request or is an error.
36 | message InferenceResult {
37 | ModelSpec model_spec = 1;
38 |
39 | oneof result {
40 | ClassificationResult classification_result = 2;
41 | RegressionResult regression_result = 3;
42 | }
43 | }
44 |
45 | // Inference request containing one or more requests.
46 | message MultiInferenceRequest {
47 | // Inference tasks.
48 | repeated InferenceTask tasks = 1;
49 |
50 | // Input data.
51 | Input input = 2;
52 | }
53 |
54 | // Inference request containing one or more responses.
55 | message MultiInferenceResponse {
56 | // List of results; one for each InferenceTask in the request, returned in the
57 | // same order as the request.
58 | repeated InferenceResult results = 1;
59 | }
60 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/graph_transfer_info.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "GraphTransferInfoProto";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 | import "tensorflow/core/framework/types.proto";
10 |
11 | message GraphTransferNodeInput {
12 | int32 node_id = 1;
13 | int32 output_port = 2;
14 | }
15 | message GraphTransferNodeInfo {
16 | string name = 1;
17 | int32 node_id = 2;
18 | string type_name = 3;
19 | int32 soc_op_id = 4;
20 | int32 padding_id = 5;
21 | int32 input_count = 6;
22 | int32 output_count = 7;
23 | };
24 | message GraphTransferConstNodeInfo {
25 | string name = 1;
26 | int32 node_id = 2;
27 | repeated int64 shape = 3;
28 | bytes data = 4;
29 | DataType dtype = 5;
30 | };
31 | message GraphTransferNodeInputInfo {
32 | int32 node_id = 1;
33 | repeated GraphTransferNodeInput node_input = 2;
34 | };
35 | message GraphTransferNodeOutputInfo {
36 | int32 node_id = 1;
37 | repeated int32 max_byte_size = 2;
38 | };
39 | message GraphTransferGraphInputNodeInfo {
40 | string name = 1;
41 | repeated int64 shape = 2;
42 | DataType dtype = 3;
43 | }
44 |
45 | message GraphTransferGraphOutputNodeInfo {
46 | string name = 1;
47 | repeated int64 shape = 2;
48 | DataType dtype = 3;
49 | }
50 |
51 | // Protocol buffer representing a handle to a tensorflow resource. Handles are
52 | // not valid across executions, but can be serialized back and forth from within
53 | // a single run.
54 | message GraphTransferInfo {
55 | enum Destination {
56 | NOP = 0;
57 | HEXAGON = 1;
58 | }
59 |
60 | repeated GraphTransferNodeInfo node_info = 1;
61 | repeated GraphTransferConstNodeInfo const_node_info = 2;
62 | repeated GraphTransferNodeInputInfo node_input_info = 3;
63 | repeated GraphTransferNodeOutputInfo node_output_info = 4;
64 | // Input Node parameters of transferred graph
65 | repeated GraphTransferGraphInputNodeInfo graph_input_node_info = 5;
66 | repeated GraphTransferGraphOutputNodeInfo graph_output_node_info = 6;
67 | // Destination of graph transfer
68 | Destination destination = 7;
69 | };
70 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/NamedTensorProtoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/named_tensor.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface NamedTensorProtoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.NamedTensorProto)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Name of the tensor. 13 | *14 | * 15 | *
string name = 1;
16 | */
17 | java.lang.String getName();
18 | /**
19 | * 20 | * Name of the tensor. 21 | *22 | * 23 | *
string name = 1;
24 | */
25 | com.google.protobuf.ByteString
26 | getNameBytes();
27 |
28 | /**
29 | * 30 | * The client can populate a TensorProto using a tensorflow::Tensor`, or 31 | * directly using the protobuf field accessors. 32 | * The client specifies whether the returned tensor values should be 33 | * filled tensor fields (float_val, int_val, etc.) or encoded in a 34 | * compact form in tensor.tensor_content. 35 | *36 | * 37 | *
.tensorflow.TensorProto tensor = 2;
38 | */
39 | boolean hasTensor();
40 | /**
41 | * 42 | * The client can populate a TensorProto using a tensorflow::Tensor`, or 43 | * directly using the protobuf field accessors. 44 | * The client specifies whether the returned tensor values should be 45 | * filled tensor fields (float_val, int_val, etc.) or encoded in a 46 | * compact form in tensor.tensor_content. 47 | *48 | * 49 | *
.tensorflow.TensorProto tensor = 2;
50 | */
51 | org.tensorflow.framework.TensorProto getTensor();
52 | /**
53 | * 54 | * The client can populate a TensorProto using a tensorflow::Tensor`, or 55 | * directly using the protobuf field accessors. 56 | * The client specifies whether the returned tensor values should be 57 | * filled tensor fields (float_val, int_val, etc.) or encoded in a 58 | * compact form in tensor.tensor_content. 59 | *60 | * 61 | *
.tensorflow.TensorProto tensor = 2;
62 | */
63 | org.tensorflow.framework.TensorProtoOrBuilder getTensorOrBuilder();
64 | }
65 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/ListDevicesResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface ListDevicesResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ListDevicesResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * repeated .tensorflow.DeviceAttributes local_device = 1;
12 | */
13 | java.util.Listrepeated .tensorflow.DeviceAttributes local_device = 1;
17 | */
18 | org.tensorflow.framework.DeviceAttributes getLocalDevice(int index);
19 | /**
20 | * repeated .tensorflow.DeviceAttributes local_device = 1;
21 | */
22 | int getLocalDeviceCount();
23 | /**
24 | * repeated .tensorflow.DeviceAttributes local_device = 1;
25 | */
26 | java.util.List
27 | getLocalDeviceOrBuilderList();
28 | /**
29 | * repeated .tensorflow.DeviceAttributes local_device = 1;
30 | */
31 | org.tensorflow.framework.DeviceAttributesOrBuilder getLocalDeviceOrBuilder(
32 | int index);
33 |
34 | /**
35 | * repeated .tensorflow.DeviceAttributes remote_device = 2;
36 | */
37 | java.util.Listrepeated .tensorflow.DeviceAttributes remote_device = 2;
41 | */
42 | org.tensorflow.framework.DeviceAttributes getRemoteDevice(int index);
43 | /**
44 | * repeated .tensorflow.DeviceAttributes remote_device = 2;
45 | */
46 | int getRemoteDeviceCount();
47 | /**
48 | * repeated .tensorflow.DeviceAttributes remote_device = 2;
49 | */
50 | java.util.List
51 | getRemoteDeviceOrBuilderList();
52 | /**
53 | * repeated .tensorflow.DeviceAttributes remote_device = 2;
54 | */
55 | org.tensorflow.framework.DeviceAttributesOrBuilder getRemoteDeviceOrBuilder(
56 | int index);
57 | }
58 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/TensorDescriptionOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/tensor_description.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface TensorDescriptionOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.TensorDescription)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Data type of tensor elements 13 | *14 | * 15 | *
.tensorflow.DataType dtype = 1;
16 | */
17 | int getDtypeValue();
18 | /**
19 | * 20 | * Data type of tensor elements 21 | *22 | * 23 | *
.tensorflow.DataType dtype = 1;
24 | */
25 | org.tensorflow.framework.DataType getDtype();
26 |
27 | /**
28 | * 29 | * Shape of the tensor. 30 | *31 | * 32 | *
.tensorflow.TensorShapeProto shape = 2;
33 | */
34 | boolean hasShape();
35 | /**
36 | * 37 | * Shape of the tensor. 38 | *39 | * 40 | *
.tensorflow.TensorShapeProto shape = 2;
41 | */
42 | org.tensorflow.framework.TensorShapeProto getShape();
43 | /**
44 | * 45 | * Shape of the tensor. 46 | *47 | * 48 | *
.tensorflow.TensorShapeProto shape = 2;
49 | */
50 | org.tensorflow.framework.TensorShapeProtoOrBuilder getShapeOrBuilder();
51 |
52 | /**
53 | * 54 | * Information about the size and allocator used for the data 55 | *56 | * 57 | *
.tensorflow.AllocationDescription allocation_description = 4;
58 | */
59 | boolean hasAllocationDescription();
60 | /**
61 | * 62 | * Information about the size and allocator used for the data 63 | *64 | * 65 | *
.tensorflow.AllocationDescription allocation_description = 4;
66 | */
67 | org.tensorflow.framework.AllocationDescription getAllocationDescription();
68 | /**
69 | * 70 | * Information about the size and allocator used for the data 71 | *72 | * 73 | *
.tensorflow.AllocationDescription allocation_description = 4;
74 | */
75 | org.tensorflow.framework.AllocationDescriptionOrBuilder getAllocationDescriptionOrBuilder();
76 | }
77 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/VariantTensorDataProtoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/tensor.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface VariantTensorDataProtoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.VariantTensorDataProto)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Name of the type of objects being serialized. 13 | *14 | * 15 | *
string type_name = 1;
16 | */
17 | java.lang.String getTypeName();
18 | /**
19 | * 20 | * Name of the type of objects being serialized. 21 | *22 | * 23 | *
string type_name = 1;
24 | */
25 | com.google.protobuf.ByteString
26 | getTypeNameBytes();
27 |
28 | /**
29 | * 30 | * Portions of the object that are not Tensors. 31 | *32 | * 33 | *
bytes metadata = 2;
34 | */
35 | com.google.protobuf.ByteString getMetadata();
36 |
37 | /**
38 | * 39 | * Tensors contained within objects being serialized. 40 | *41 | * 42 | *
repeated .tensorflow.TensorProto tensors = 3;
43 | */
44 | java.util.List48 | * Tensors contained within objects being serialized. 49 | *50 | * 51 | *
repeated .tensorflow.TensorProto tensors = 3;
52 | */
53 | org.tensorflow.framework.TensorProto getTensors(int index);
54 | /**
55 | * 56 | * Tensors contained within objects being serialized. 57 | *58 | * 59 | *
repeated .tensorflow.TensorProto tensors = 3;
60 | */
61 | int getTensorsCount();
62 | /**
63 | * 64 | * Tensors contained within objects being serialized. 65 | *66 | * 67 | *
repeated .tensorflow.TensorProto tensors = 3;
68 | */
69 | java.util.List
70 | getTensorsOrBuilderList();
71 | /**
72 | * 73 | * Tensors contained within objects being serialized. 74 | *75 | * 76 | *
repeated .tensorflow.TensorProto tensors = 3;
77 | */
78 | org.tensorflow.framework.TensorProtoOrBuilder getTensorsOrBuilder(
79 | int index);
80 | }
81 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/ExtendSessionRequestOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/master.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface ExtendSessionRequestOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ExtendSessionRequest)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * REQUIRED: session_handle must be returned by a CreateSession call 13 | * to the same master service. 14 | *15 | * 16 | *
string session_handle = 1;
17 | */
18 | java.lang.String getSessionHandle();
19 | /**
20 | * 21 | * REQUIRED: session_handle must be returned by a CreateSession call 22 | * to the same master service. 23 | *24 | * 25 | *
string session_handle = 1;
26 | */
27 | com.google.protobuf.ByteString
28 | getSessionHandleBytes();
29 |
30 | /**
31 | * 32 | * REQUIRED: The nodes to be added to the session's graph. If any node has 33 | * the same name as an existing node, the operation will fail with 34 | * ILLEGAL_ARGUMENT. 35 | *36 | * 37 | *
.tensorflow.GraphDef graph_def = 2;
38 | */
39 | boolean hasGraphDef();
40 | /**
41 | * 42 | * REQUIRED: The nodes to be added to the session's graph. If any node has 43 | * the same name as an existing node, the operation will fail with 44 | * ILLEGAL_ARGUMENT. 45 | *46 | * 47 | *
.tensorflow.GraphDef graph_def = 2;
48 | */
49 | org.tensorflow.framework.GraphDef getGraphDef();
50 | /**
51 | * 52 | * REQUIRED: The nodes to be added to the session's graph. If any node has 53 | * the same name as an existing node, the operation will fail with 54 | * ILLEGAL_ARGUMENT. 55 | *56 | * 57 | *
.tensorflow.GraphDef graph_def = 2;
58 | */
59 | org.tensorflow.framework.GraphDefOrBuilder getGraphDefOrBuilder();
60 |
61 | /**
62 | * 63 | * REQUIRED: The version number of the graph to be extended. This will be 64 | * tested against the current server-side version number, and the operation 65 | * will fail with FAILED_PRECONDITION if they do not match. 66 | *67 | * 68 | *
int64 current_graph_version = 3;
69 | */
70 | long getCurrentGraphVersion();
71 | }
72 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/CompleteGroupResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface CompleteGroupResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.CompleteGroupResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * int32 group_key = 1;
12 | */
13 | int getGroupKey();
14 |
15 | /**
16 | * int32 group_size = 2;
17 | */
18 | int getGroupSize();
19 |
20 | /**
21 | * string device_type = 3;
22 | */
23 | java.lang.String getDeviceType();
24 | /**
25 | * string device_type = 3;
26 | */
27 | com.google.protobuf.ByteString
28 | getDeviceTypeBytes();
29 |
30 | /**
31 | * 32 | * number of distinct tasks hosting the devices 33 | *34 | * 35 | *
int32 num_tasks = 4;
36 | */
37 | int getNumTasks();
38 |
39 | /**
40 | * repeated string device_name = 5;
41 | */
42 | java.util.Listrepeated string device_name = 5;
46 | */
47 | int getDeviceNameCount();
48 | /**
49 | * repeated string device_name = 5;
50 | */
51 | java.lang.String getDeviceName(int index);
52 | /**
53 | * repeated string device_name = 5;
54 | */
55 | com.google.protobuf.ByteString
56 | getDeviceNameBytes(int index);
57 |
58 | /**
59 | * 60 | * task name prefixes of device_names 61 | *62 | * 63 | *
repeated string task_name = 6;
64 | */
65 | java.util.List69 | * task name prefixes of device_names 70 | *71 | * 72 | *
repeated string task_name = 6;
73 | */
74 | int getTaskNameCount();
75 | /**
76 | * 77 | * task name prefixes of device_names 78 | *79 | * 80 | *
repeated string task_name = 6;
81 | */
82 | java.lang.String getTaskName(int index);
83 | /**
84 | * 85 | * task name prefixes of device_names 86 | *87 | * 88 | *
repeated string task_name = 6;
89 | */
90 | com.google.protobuf.ByteString
91 | getTaskNameBytes(int index);
92 | }
93 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/distruntime/RecvTensorResponseOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/protobuf/worker.proto
3 |
4 | package org.tensorflow.distruntime;
5 |
6 | public interface RecvTensorResponseOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.RecvTensorResponse)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * The tensor as a proto. 13 | *14 | * 15 | *
.tensorflow.TensorProto tensor = 1;
16 | */
17 | boolean hasTensor();
18 | /**
19 | * 20 | * The tensor as a proto. 21 | *22 | * 23 | *
.tensorflow.TensorProto tensor = 1;
24 | */
25 | org.tensorflow.framework.TensorProto getTensor();
26 | /**
27 | * 28 | * The tensor as a proto. 29 | *30 | * 31 | *
.tensorflow.TensorProto tensor = 1;
32 | */
33 | org.tensorflow.framework.TensorProtoOrBuilder getTensorOrBuilder();
34 |
35 | /**
36 | * 37 | * If true, this tensor was the output of a dead node, and the 38 | * content is invalid. 39 | *40 | * 41 | *
bool is_dead = 2;
42 | */
43 | boolean getIsDead();
44 |
45 | /**
46 | * 47 | * The time at which tensor was available and started to be returned. 48 | *49 | * 50 | *
int64 send_start_micros = 3;
51 | */
52 | long getSendStartMicros();
53 |
54 | /**
55 | * 56 | * Optional additional information about how to receive the tensor, 57 | * e.g. in the event that `RecvTensorRequest.dma_ok` was true. 58 | *59 | * 60 | *
.google.protobuf.Any transport_options = 4;
61 | */
62 | boolean hasTransportOptions();
63 | /**
64 | * 65 | * Optional additional information about how to receive the tensor, 66 | * e.g. in the event that `RecvTensorRequest.dma_ok` was true. 67 | *68 | * 69 | *
.google.protobuf.Any transport_options = 4;
70 | */
71 | com.google.protobuf.Any getTransportOptions();
72 | /**
73 | * 74 | * Optional additional information about how to receive the tensor, 75 | * e.g. in the event that `RecvTensorRequest.dma_ok` was true. 76 | *77 | * 78 | *
.google.protobuf.Any transport_options = 4;
79 | */
80 | com.google.protobuf.AnyOrBuilder getTransportOptionsOrBuilder();
81 | }
82 |
--------------------------------------------------------------------------------
/target/generated-sources/protobuf/java/org/tensorflow/framework/ResourceHandleProtoOrBuilder.java:
--------------------------------------------------------------------------------
1 | // Generated by the protocol buffer compiler. DO NOT EDIT!
2 | // source: tensorflow/core/framework/resource_handle.proto
3 |
4 | package org.tensorflow.framework;
5 |
6 | public interface ResourceHandleProtoOrBuilder extends
7 | // @@protoc_insertion_point(interface_extends:tensorflow.ResourceHandleProto)
8 | com.google.protobuf.MessageOrBuilder {
9 |
10 | /**
11 | * 12 | * Unique name for the device containing the resource. 13 | *14 | * 15 | *
string device = 1;
16 | */
17 | java.lang.String getDevice();
18 | /**
19 | * 20 | * Unique name for the device containing the resource. 21 | *22 | * 23 | *
string device = 1;
24 | */
25 | com.google.protobuf.ByteString
26 | getDeviceBytes();
27 |
28 | /**
29 | * 30 | * Container in which this resource is placed. 31 | *32 | * 33 | *
string container = 2;
34 | */
35 | java.lang.String getContainer();
36 | /**
37 | * 38 | * Container in which this resource is placed. 39 | *40 | * 41 | *
string container = 2;
42 | */
43 | com.google.protobuf.ByteString
44 | getContainerBytes();
45 |
46 | /**
47 | * 48 | * Unique name of this resource. 49 | *50 | * 51 | *
string name = 3;
52 | */
53 | java.lang.String getName();
54 | /**
55 | * 56 | * Unique name of this resource. 57 | *58 | * 59 | *
string name = 3;
60 | */
61 | com.google.protobuf.ByteString
62 | getNameBytes();
63 |
64 | /**
65 | * 66 | * Hash code for the type of the resource. Is only valid in the same device 67 | * and in the same execution. 68 | *69 | * 70 | *
uint64 hash_code = 4;
71 | */
72 | long getHashCode();
73 |
74 | /**
75 | * 76 | * For debug-only, the name of the type pointed to by this handle, if 77 | * available. 78 | *79 | * 80 | *
string maybe_type_name = 5;
81 | */
82 | java.lang.String getMaybeTypeName();
83 | /**
84 | * 85 | * For debug-only, the name of the type pointed to by this handle, if 86 | * available. 87 | *88 | * 89 | *
string maybe_type_name = 5;
90 | */
91 | com.google.protobuf.ByteString
92 | getMaybeTypeNameBytes();
93 | }
94 |
--------------------------------------------------------------------------------
/src/main/proto/tensorflow/core/framework/graph.proto:
--------------------------------------------------------------------------------
1 | syntax = "proto3";
2 |
3 | package tensorflow;
4 | option cc_enable_arenas = true;
5 | option java_outer_classname = "GraphProtos";
6 | option java_multiple_files = true;
7 | option java_package = "org.tensorflow.framework";
8 | option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9 | import "tensorflow/core/framework/node_def.proto";
10 | import "tensorflow/core/framework/function.proto";
11 | import "tensorflow/core/framework/versions.proto";
12 |
13 | // Represents the graph of operations
14 | message GraphDef {
15 | repeated NodeDef node = 1;
16 |
17 | // Compatibility versions of the graph. See core/public/version.h for version
18 | // history. The GraphDef version is distinct from the TensorFlow version, and
19 | // each release of TensorFlow will support a range of GraphDef versions.
20 | VersionDef versions = 4;
21 |
22 | // Deprecated single version field; use versions above instead. Since all
23 | // GraphDef changes before "versions" was introduced were forward
24 | // compatible, this field is entirely ignored.
25 | int32 version = 3 [deprecated = true];
26 |
27 | // EXPERIMENTAL. DO NOT USE OR DEPEND ON THIS YET.
28 | //
29 | // "library" provides user-defined functions.
30 | //
31 | // Naming:
32 | // * library.function.name are in a flat namespace.
33 | // NOTE: We may need to change it to be hierarchical to support
34 | // different orgs. E.g.,
35 | // { "/google/nn", { ... }},
36 | // { "/google/vision", { ... }}
37 | // { "/org_foo/module_bar", { ... }}
38 | // map