experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Give unique go package paths in core/framework to avoid circular dependencies.

Browse Source 
PiperOrigin-RevId: 293433590
Change-Id: Ia0dd6ed36cadcf2372805f6a83d5cfb51618db8c
This commit is contained in:
Jonathan Hseu 2020-02-05 13:07:47 -08:00 committed by TensorFlower Gardener
parent c6f7b48a3a
commit 4221d1aa4d
60 changed files with 248 additions and 208 deletions

10
tensorflow/core/example/example.proto
View File

@ -2,13 +2,15 @@
// model training or inference.
syntax = "proto3";
package tensorflow;
import "tensorflow/core/example/feature.proto";
option cc_enable_arenas = true;
option java_outer_classname = "ExampleProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.example";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/example";
package tensorflow;
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/example/example_protos_go_proto";
// An Example is a mostly-normalized data format for storing data for
// training and inference. It contains a key-value store (features); where
@ -87,7 +89,7 @@ package tensorflow;
message Example {
Features features = 1;
};
}
// A SequenceExample is an Example representing one or more sequences, and
// some context. The context contains features which apply to the entire
@ -298,4 +300,4 @@ message Example {
message SequenceExample {
Features context = 1;
FeatureLists feature_lists = 2;
};
}

21
tensorflow/core/example/example_parser_configuration.proto
View File

@ -2,38 +2,39 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/tensor.proto";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option cc_enable_arenas = true;
option java_outer_classname = "ExampleParserConfigurationProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.example";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/example";
package tensorflow;
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/tensor.proto";
import "tensorflow/core/framework/types.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/example/example_parser_configuration_go_proto";
message VarLenFeatureProto {
tensorflow.DataType dtype = 1;
string values_output_tensor_name = 2;
string indices_output_tensor_name = 3;
string shapes_output_tensor_name = 4;
};
}
message FixedLenFeatureProto {
tensorflow.DataType dtype = 1;
tensorflow.TensorShapeProto shape = 2;
tensorflow.TensorProto default_value = 3;
string values_output_tensor_name = 4;
};
}
message FeatureConfiguration {
oneof config {
FixedLenFeatureProto fixed_len_feature = 1;
VarLenFeatureProto var_len_feature = 2;
}
};
}
message ExampleParserConfiguration {
map<string, FeatureConfiguration> feature_map = 1;
};
}

14
tensorflow/core/example/feature.proto
View File

@ -54,12 +54,14 @@
//
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "FeatureProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.example";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/example";
package tensorflow;
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/example/example_protos_go_proto";
// Containers to hold repeated fundamental values.
message BytesList {
@ -80,12 +82,12 @@ message Feature {
FloatList float_list = 2;
Int64List int64_list = 3;
}
};
}
message Features {
// Map from feature name to feature.
map<string, Feature> feature = 1;
};
}
// Containers for sequential data.
//
@ -97,9 +99,9 @@ message Features {
//
message FeatureList {
repeated Feature feature = 1;
};
}
message FeatureLists {
// Map from feature name to feature list.
map<string, FeatureList> feature_list = 1;
};
}

5
tensorflow/core/framework/allocation_description.proto
View File

@ -1,11 +1,12 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "AllocationDescriptionProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/allocation_description_go_proto";
message AllocationDescription {
// Total number of bytes requested
@ -25,4 +26,4 @@ message AllocationDescription {
// Address of the allocation.
uint64 ptr = 6;
};
}

10
tensorflow/core/framework/attr_value.proto
View File

@ -1,14 +1,16 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/tensor.proto";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option cc_enable_arenas = true;
option java_outer_classname = "AttrValueProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/tensor.proto";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/attr_value_go_proto";
// Protocol buffer representing the value for an attr used to configure an Op.
// Comment indicates the corresponding attr type. Only the field matching the

8
tensorflow/core/framework/cost_graph.proto
View File

@ -1,13 +1,15 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option cc_enable_arenas = true;
option java_outer_classname = "CostGraphProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/cost_graph_go_proto";
message CostGraphDef {
message Node {

9
tensorflow/core/framework/device_attributes.proto
View File

@ -1,21 +1,22 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "DeviceAttributesProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/device_attributes_go_proto";
message InterconnectLink {
int32 device_id = 1;
string type = 2;
int32 strength = 3;
};
}
message LocalLinks {
repeated InterconnectLink link = 1;
};
}
message DeviceLocality {
// Optional bus locality of device. Default value of 0 means
@ -27,7 +28,7 @@ message DeviceLocality {
// Optional local interconnect links to other devices.
LocalLinks links = 3;
};
}
message DeviceAttributes {
// Fully specified name of the device within a cluster.

10
tensorflow/core/framework/function.proto
View File

@ -1,14 +1,16 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/attr_value.proto";
import "tensorflow/core/framework/node_def.proto";
import "tensorflow/core/framework/op_def.proto";
option cc_enable_arenas = true;
option java_outer_classname = "FunctionProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/attr_value.proto";
import "tensorflow/core/framework/node_def.proto";
import "tensorflow/core/framework/op_def.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/function_go_proto";
// A library is a set of named functions.
message FunctionDefLibrary {

12
tensorflow/core/framework/graph.proto
View File

@ -1,14 +1,16 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/function.proto";
import "tensorflow/core/framework/node_def.proto";
import "tensorflow/core/framework/versions.proto";
option cc_enable_arenas = true;
option java_outer_classname = "GraphProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/node_def.proto";
import "tensorflow/core/framework/function.proto";
import "tensorflow/core/framework/versions.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/graph_go_proto";
// Represents the graph of operations
message GraphDef {
@ -53,4 +55,4 @@ message GraphDef {
// consumer does not start until all return values of the callee
// function are ready.
FunctionDefLibrary library = 2;
};
}

16
tensorflow/core/framework/graph_transfer_info.proto
View File

@ -1,12 +1,14 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/types.proto";
option cc_enable_arenas = true;
option java_outer_classname = "GraphTransferInfoProto";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/types.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/graph_transfer_info_go_proto";
message GraphTransferNodeInput {
int32 node_id = 1;
@ -20,22 +22,22 @@ message GraphTransferNodeInfo {
int32 padding_id = 5;
int32 input_count = 6;
int32 output_count = 7;
};
}
message GraphTransferConstNodeInfo {
string name = 1;
int32 node_id = 2;
repeated int64 shape = 3;
bytes data = 4;
DataType dtype = 5;
};
}
message GraphTransferNodeInputInfo {
int32 node_id = 1;
repeated GraphTransferNodeInput node_input = 2;
};
}
message GraphTransferNodeOutputInfo {
int32 node_id = 1;
repeated int32 max_byte_size = 2;
};
}
message GraphTransferGraphInputNodeInfo {
string name = 1;
repeated int64 shape = 2;
@ -66,4 +68,4 @@ message GraphTransferInfo {
repeated GraphTransferGraphOutputNodeInfo graph_output_node_info = 6;
// Destination of graph transfer
Destination destination = 7;
};
}

8
tensorflow/core/framework/kernel_def.proto
View File

@ -1,12 +1,14 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/attr_value.proto";
option cc_enable_arenas = true;
option java_outer_classname = "KernelDefProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/attr_value.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/kernel_def_go_proto";
message KernelDef {
// Must match the name of an Op.
@ -43,4 +45,4 @@ message KernelDef {
// A collection of KernelDefs
message KernelList {
repeated KernelDef kernel = 1;
};
}

16
tensorflow/core/framework/log_memory.proto
View File

@ -1,12 +1,14 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/tensor_description.proto";
option cc_enable_arenas = true;
option java_outer_classname = "LogMemoryProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/tensor_description.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/log_memory_go_proto";
message MemoryLogStep {
// Process-unique step id.
@ -14,7 +16,7 @@ message MemoryLogStep {
// Handle describing the feeds and fetches of the step.
string handle = 2;
};
}
message MemoryLogTensorAllocation {
// Process-unique step id.
@ -26,7 +28,7 @@ message MemoryLogTensorAllocation {
// Allocated tensor details.
TensorDescription tensor = 3;
};
}
message MemoryLogTensorDeallocation {
// Id of the tensor buffer being deallocated, used to match to a
@ -35,7 +37,7 @@ message MemoryLogTensorDeallocation {
// Name of the allocator used.
string allocator_name = 2;
};
}
message MemoryLogTensorOutput {
// Process-unique step id.
@ -71,7 +73,7 @@ message MemoryLogRawAllocation {
// Name of the allocator used.
string allocator_name = 6;
};
}
message MemoryLogRawDeallocation {
// Process-unique step id.
@ -90,4 +92,4 @@ message MemoryLogRawDeallocation {
// True if the deallocation is queued and will be performed later,
// e.g. for GPU lazy freeing of buffers.
bool deferred = 5;
};
}

10
tensorflow/core/framework/node_def.proto
View File

@ -1,12 +1,14 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/attr_value.proto";
option cc_enable_arenas = true;
option java_outer_classname = "NodeProto";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/attr_value.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/node_def_go_proto";
message NodeDef {
// The name given to this operator. Used for naming inputs,
@ -79,8 +81,8 @@ message NodeDef {
// `original_node_names` can be used to map errors originating at the
// current ndoe to some top level source code.
repeated string original_func_names = 2;
};
}
// This stores debug information associated with the node.
ExperimentalDebugInfo experimental_debug_info = 6;
};
}

5
tensorflow/core/framework/reader_base.proto
View File

@ -1,11 +1,12 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "ReaderBaseProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/reader_base_go_proto";
// For serializing and restoring the state of ReaderBase, see
// reader_base.h for details.
@ -14,4 +15,4 @@ message ReaderBaseState {
int64 work_finished = 2;
int64 num_records_produced = 3;
bytes current_work = 4;
};
}

13
tensorflow/core/framework/remote_fused_graph_execute_info.proto
View File

@ -1,20 +1,21 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/graph.proto";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option cc_enable_arenas = true;
option java_outer_classname = "RemoteFusedGraphExecuteInfoProto";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/graph.proto";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/remote_fused_graph_execute_info_go_proto";
// Protocol buffer representing a handle to a tensorflow resource. Handles are
// not valid across executions, but can be serialized back and forth from within
// a single run.
message RemoteFusedGraphExecuteInfo {
message TensorShapeTypeProto {
DataType dtype = 1;
TensorShapeProto shape = 2;
@ -44,4 +45,4 @@ message RemoteFusedGraphExecuteInfo {
// TODO(satok): Remote output tensor shape once shape information is stored
// in NodeDef
repeated TensorShapeTypeProto default_graph_output_tensor_shape = 7;
};
}

11
tensorflow/core/framework/resource_handle.proto
View File

@ -1,14 +1,15 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option cc_enable_arenas = true;
option java_outer_classname = "ResourceHandle";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/resource_handle_go_proto";
// Protocol buffer representing a handle to a tensorflow resource. Handles are
// not valid across executions, but can be serialized back and forth from within
@ -39,4 +40,4 @@ message ResourceHandleProto {
// Data types and shapes for the underlying resource.
repeated DtypeAndShape dtypes_and_shapes = 6;
};
}

14
tensorflow/core/framework/step_stats.proto
View File

@ -1,13 +1,15 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/allocation_description.proto";
import "tensorflow/core/framework/tensor_description.proto";
option cc_enable_arenas = true;
option java_outer_classname = "StepStatsProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/allocation_description.proto";
import "tensorflow/core/framework/tensor_description.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/step_stats_go_proto";
// An allocation/de-allocation operation performed by the allocator.
message AllocationRecord {
@ -36,7 +38,7 @@ message AllocatorMemoryUsed {
message NodeOutput {
int32 slot = 1;
TensorDescription tensor_description = 3;
};
}
// For memory tracking.
message MemoryStats {
@ -72,7 +74,7 @@ message NodeExecStats {
int64 op_end_rel_nanos = 15;
int64 all_end_rel_nanos = 16;
int64 scheduled_nanos = 17;
};
}
message DeviceStepStats {
string device = 1;
@ -83,4 +85,4 @@ message DeviceStepStats {
message StepStats {
repeated DeviceStepStats dev_stats = 1;
};
}

10
tensorflow/core/framework/summary.proto
View File

@ -1,12 +1,14 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/tensor.proto";
option cc_enable_arenas = true;
option java_outer_classname = "SummaryProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/tensor.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/summary_go_proto";
// Metadata associated with a series of Summary data
message SummaryDescription {
@ -31,7 +33,7 @@ message HistogramProto {
// i != 0: bucket_limit(i-1) .. bucket_limit(i)
repeated double bucket_limit = 6 [packed = true];
repeated double bucket = 7 [packed = true];
};
}
// A SummaryMetadata encapsulates information on which plugins are able to make
// use of a certain summary value.
@ -59,7 +61,7 @@ message SummaryMetadata {
// imposes constraints on the dtype and shape of the corresponding tensor
// values. See `DataClass` docs for details.
DataClass data_class = 4;
};
}
enum DataClass {
// Unknown data class, used (implicitly) for legacy data. Will not be

12
tensorflow/core/framework/tensor.proto
View File

@ -1,14 +1,16 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/resource_handle.proto";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option cc_enable_arenas = true;
option java_outer_classname = "TensorProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/resource_handle.proto";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_go_proto";
// Protocol buffer representing a tensor.
message TensorProto {
@ -81,7 +83,7 @@ message TensorProto {
// DT_UINT64
repeated uint64 uint64_val = 17 [packed = true];
};
}
// Protocol buffer representing the serialization format of DT_VARIANT tensors.
message VariantTensorDataProto {

12
tensorflow/core/framework/tensor_description.proto
View File

@ -1,14 +1,16 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/allocation_description.proto";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option cc_enable_arenas = true;
option java_outer_classname = "TensorDescriptionProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
import "tensorflow/core/framework/types.proto";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/allocation_description.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_description_go_proto";
message TensorDescription {
// Data type of tensor elements
@ -19,4 +21,4 @@ message TensorDescription {
// Information about the size and allocator used for the data
AllocationDescription allocation_description = 4;
};
}

11
tensorflow/core/framework/tensor_slice.proto
View File

@ -1,13 +1,14 @@
// Protocol buffer representing slices of a tensor
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "TensorSliceProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
package tensorflow;
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_slice_go_proto";
// Can only be interpreted if you know the corresponding TensorShape.
message TensorSliceProto {
@ -27,7 +28,7 @@ message TensorSliceProto {
oneof has_length {
int64 length = 2;
}
};
}
// Extent of the slice in all tensor dimensions.
//
@ -35,4 +36,4 @@ message TensorSliceProto {
// slice belongs to. The order of sizes is the same as the order of
// dimensions in the TensorShape.
repeated Extent extent = 1;
};
}

3
tensorflow/core/framework/variable.proto
View File

@ -6,8 +6,7 @@ option cc_enable_arenas = true;
option java_outer_classname = "VariableProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/variable_go_proto";
// Indicates when a distributed variable will be synced.
enum VariableSynchronization {

5
tensorflow/core/framework/versions.proto
View File

@ -1,11 +1,12 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "VersionsProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/versions_go_proto";
// Version information for a piece of serialized data
//
@ -29,4 +30,4 @@ message VersionDef {
// Specific consumer versions which are disallowed (e.g. due to bugs).
repeated int32 bad_consumers = 3;
};
}

2
tensorflow/core/protobuf/autotuning.proto
View File

@ -10,7 +10,7 @@ package tensorflow;
import "google/protobuf/any.proto";
import "google/protobuf/duration.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
message CudnnVersion {
int32 major = 1;

2
tensorflow/core/protobuf/bfc_memory_map.proto
View File

@ -2,7 +2,7 @@ syntax = "proto3";
package tensorflow;
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Some of the data from AllocatorStats
message MemAllocatorStats {

3
tensorflow/core/protobuf/cluster.proto
View File

@ -16,11 +16,12 @@ limitations under the License.
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "ClusterProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.distruntime";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// This file contains protos to be used when defining a TensorFlow
// cluster.

16
tensorflow/core/protobuf/config.proto
View File

@ -2,12 +2,6 @@ syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "ConfigProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/framework/cost_graph.proto";
import "tensorflow/core/framework/graph.proto";
import "tensorflow/core/framework/step_stats.proto";
@ -15,6 +9,12 @@ import "tensorflow/core/protobuf/cluster.proto";
import "tensorflow/core/protobuf/debug.proto";
import "tensorflow/core/protobuf/rewriter_config.proto";
option cc_enable_arenas = true;
option java_outer_classname = "ConfigProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
message GPUOptions {
// Fraction of the available GPU memory to allocate for each process.
// 1 means to allocate all of the GPU memory, 0.5 means the process
@ -578,7 +578,7 @@ message ConfigProto {
// The XLA fusion autotuner can improve performance by executing a heuristic
// search on the compiler parameters.
int64 xla_fusion_autotuner_thresh = 15;
};
}
Experimental experimental = 16;
@ -636,7 +636,7 @@ message RunOptions {
// and tail) latency.
// Consider using this option for CPU-bound workloads like inference.
bool use_run_handler_pool = 2;
};
}
Experimental experimental = 8;

3
tensorflow/core/protobuf/control_flow.proto
View File

@ -1,11 +1,12 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "ControlFlowProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Control flow context related protocol buffers.

2
tensorflow/core/protobuf/conv_autotuning.proto
View File

@ -6,7 +6,7 @@ package tensorflow;
import "tensorflow/stream_executor/dnn.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// A convolution. Currently it's only used for logging. In the future, we may
// want to use it in the API as well.

3
tensorflow/core/protobuf/critical_section.proto
View File

@ -1,11 +1,12 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "CriticalSectionProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Protocol buffer representing a CriticalSection.
message CriticalSectionDef {

3
tensorflow/core/protobuf/debug.proto
View File

@ -1,11 +1,12 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "DebugProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Option for watching a node in TensorFlow Debugger (tfdbg).
message DebugTensorWatch {

3
tensorflow/core/protobuf/debug_event.proto
View File

@ -9,8 +9,7 @@ option cc_enable_arenas = true;
option java_outer_classname = "DebugEventProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.util";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Available modes for extracting debugging information from a Tensor.
// TODO(cais): Document the detailed column names and semantics in a separate

3
tensorflow/core/protobuf/device_filters.proto
View File

@ -21,8 +21,7 @@ option cc_enable_arenas = true;
option java_outer_classname = "DeviceFiltersProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.distruntime";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// This file contains protos to be used when defining a TensorFlow
// cluster.

3
tensorflow/core/protobuf/device_properties.proto
View File

@ -16,9 +16,10 @@ limitations under the License.
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "DevicePropertiesProtos";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
message DeviceProperties {
// Device type (CPU, GPU, ...)

2
tensorflow/core/protobuf/eager_service.proto
View File

@ -11,7 +11,7 @@ import "tensorflow/core/framework/versions.proto";
import "tensorflow/core/protobuf/remote_tensor_handle.proto";
import "tensorflow/core/protobuf/tensorflow_server.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// A proto representation of an eager operation.
message Operation {

3
tensorflow/core/protobuf/error_codes.proto
View File

@ -6,8 +6,7 @@ option cc_enable_arenas = true;
option java_outer_classname = "ErrorCodesProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// The canonical error codes for TensorFlow APIs.
//

3
tensorflow/core/protobuf/graph_debug_info.proto
View File

@ -6,8 +6,7 @@ option cc_enable_arenas = true;
option java_outer_classname = "GraphDebugInfoProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
message GraphDebugInfo {
// This represents a file/line location in the source code.

12
tensorflow/core/protobuf/master.proto
View File

@ -17,12 +17,6 @@ syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "DistributedRuntimeProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.distruntime";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/framework/device_attributes.proto";
import "tensorflow/core/framework/graph.proto";
import "tensorflow/core/framework/tensor.proto";
@ -30,6 +24,12 @@ import "tensorflow/core/protobuf/config.proto";
import "tensorflow/core/protobuf/error_codes.proto";
import "tensorflow/core/protobuf/named_tensor.proto";
option cc_enable_arenas = true;
option java_outer_classname = "DistributedRuntimeProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.distruntime";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
////////////////////////////////////////////////////////////////////////////////
//
// CreateSession method request/response protos.

6
tensorflow/core/protobuf/master_service.proto
View File

@ -16,11 +16,13 @@ limitations under the License.
syntax = "proto3";
package tensorflow.grpc;
import "tensorflow/core/protobuf/master.proto";
option java_outer_classname = "MasterServiceProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.distruntime";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/protobuf/master.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
////////////////////////////////////////////////////////////////////////////////
//

12
tensorflow/core/protobuf/meta_graph.proto
View File

@ -2,12 +2,6 @@ syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "MetaGraphProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "google/protobuf/any.proto";
import "tensorflow/core/framework/graph.proto";
import "tensorflow/core/framework/op_def.proto";
@ -17,6 +11,12 @@ import "tensorflow/core/protobuf/saved_object_graph.proto";
import "tensorflow/core/protobuf/saver.proto";
import "tensorflow/core/protobuf/struct.proto";
option cc_enable_arenas = true;
option java_outer_classname = "MetaGraphProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// NOTE: This protocol buffer is evolving, and will go through revisions in the
// coming months.
//

6
tensorflow/core/protobuf/named_tensor.proto
View File

@ -1,12 +1,14 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/tensor.proto";
option cc_enable_arenas = true;
option java_outer_classname = "NamedTensorProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/framework/tensor.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// A pair of tensor name and tensor values.
message NamedTensorProto {

6
tensorflow/core/protobuf/queue_runner.proto
View File

@ -2,13 +2,13 @@ syntax = "proto3";
package tensorflow;
import "tensorflow/core/protobuf/error_codes.proto";
option cc_enable_arenas = true;
option java_outer_classname = "QueueRunnerProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/protobuf/error_codes.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Protocol buffer representing a QueueRunner.
message QueueRunnerDef {

3
tensorflow/core/protobuf/remote_tensor_handle.proto
View File

@ -9,8 +9,7 @@ option cc_enable_arenas = true;
option java_outer_classname = "RemoteTensorHandleProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
message ResourceDtypeAndShape {
DataType dtype = 1;

3
tensorflow/core/protobuf/replay_log.proto
View File

@ -5,8 +5,7 @@ package tensorflow;
import "tensorflow/core/protobuf/master.proto";
option cc_enable_arenas = true;
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Records the creation of a new replay session. We record the device listing
// here to capture the state of the cluster.

9
tensorflow/core/protobuf/rewriter_config.proto
View File

@ -2,15 +2,14 @@ syntax = "proto3";
package tensorflow;
import "tensorflow/core/framework/attr_value.proto";
import "tensorflow/core/protobuf/verifier_config.proto";
option cc_enable_arenas = true;
option java_outer_classname = "RewriterConfigProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/framework/attr_value.proto";
import "tensorflow/core/protobuf/verifier_config.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
message AutoParallelOptions {
bool enable = 1;

6
tensorflow/core/protobuf/saved_model.proto
View File

@ -1,12 +1,14 @@
syntax = "proto3";
package tensorflow;
import "tensorflow/core/protobuf/meta_graph.proto";
option cc_enable_arenas = true;
option java_outer_classname = "SavedModelProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/protobuf/meta_graph.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// SavedModel is the high level serialization format for TensorFlow Models.
// See [todo: doc links, similar to session_bundle] for more information.

3
tensorflow/core/protobuf/saved_object_graph.proto
View File

@ -10,8 +10,7 @@ import "tensorflow/core/protobuf/struct.proto";
import "tensorflow/core/protobuf/trackable_object_graph.proto";
option cc_enable_arenas = true;
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// A SavedObjectGraph is part of object-based SavedModels in TF 2.0. It
// describes the directed graph of Python objects (or equivalent in other

3
tensorflow/core/protobuf/saver.proto
View File

@ -1,11 +1,12 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "SaverProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.util";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Protocol buffer representing the configuration of a Saver.
message SaverDef {

4
tensorflow/core/protobuf/struct.proto
View File

@ -2,11 +2,11 @@ syntax = "proto3";
package tensorflow;
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/types.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// `StructuredValue` represents a dynamically typed value representing various
// data structures that are inspired by Python data structures typically used in
// TensorFlow functions as inputs and outputs.

12
tensorflow/core/protobuf/tensor_bundle.proto
View File

@ -1,16 +1,18 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "TensorBundleProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.util";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/framework/tensor_shape.proto";
import "tensorflow/core/framework/tensor_slice.proto";
import "tensorflow/core/framework/types.proto";
import "tensorflow/core/framework/versions.proto";
option cc_enable_arenas = true;
option java_outer_classname = "TensorBundleProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.util";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Protos used in the tensor bundle module (tf/core/util/tensor_bundle/).
// Special header that is associated with a bundle.

2
tensorflow/core/protobuf/tensorflow_server.proto
View File

@ -25,8 +25,8 @@ option cc_enable_arenas = true;
option java_outer_classname = "ServerProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.distruntime";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
// Defines the configuration of a single TensorFlow server.
message ServerDef {
// The cluster of which this server is a member.

3
tensorflow/core/protobuf/trace_events.proto
View File

@ -6,8 +6,7 @@ option cc_enable_arenas = true;
option java_outer_classname = "TraceEventsProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// A 'Trace' contains metadata for the individual traces of a system.
message Trace {

3
tensorflow/core/protobuf/trackable_object_graph.proto
View File

@ -3,8 +3,7 @@ syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// A TensorBundle addition which saves extra information about the objects which
// own variables, allowing for more robust checkpoint loading into modified

2
tensorflow/core/protobuf/transport_options.proto
View File

@ -2,7 +2,7 @@ syntax = "proto3";
package tensorflow;
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// Extra data needed on a non-RDMA RecvBufResponse.
message RecvBufRespExtra {

3
tensorflow/core/protobuf/verifier_config.proto
View File

@ -1,11 +1,12 @@
syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "VerifierConfigProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.framework";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
// The config for graph verifiers.
message VerifierConfig {

12
tensorflow/core/protobuf/worker.proto
View File

@ -17,12 +17,6 @@ syntax = "proto3";
package tensorflow;
option cc_enable_arenas = true;
option java_outer_classname = "WorkerProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.distruntime";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "google/protobuf/any.proto";
import "tensorflow/core/framework/cost_graph.proto";
import "tensorflow/core/framework/device_attributes.proto";
@ -37,6 +31,12 @@ import "tensorflow/core/protobuf/error_codes.proto";
import "tensorflow/core/protobuf/named_tensor.proto";
import "tensorflow/core/protobuf/tensorflow_server.proto";
option cc_enable_arenas = true;
option java_outer_classname = "WorkerProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.distruntime";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
////////////////////////////////////////////////////////////////////////////////
//
// GetStatus method request/response messages

9
tensorflow/core/protobuf/worker_service.proto
View File

@ -16,11 +16,13 @@ limitations under the License.
syntax = "proto3";
package tensorflow.grpc;
import "tensorflow/core/protobuf/worker.proto";
option java_outer_classname = "WorkerServiceProtos";
option java_multiple_files = true;
option java_package = "org.tensorflow.distruntime";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf";
import "tensorflow/core/protobuf/worker.proto";
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto";
////////////////////////////////////////////////////////////////////////////////
//
@ -74,8 +76,7 @@ service WorkerService {
rpc Tracing(TracingRequest) returns (TracingResponse);
// See worker.proto for details.
rpc RecvBuf(RecvBufRequest) returns (RecvBufResponse) {
}
rpc RecvBuf(RecvBufRequest) returns (RecvBufResponse) {}
// See worker.proto for details.
rpc GetStepSequence(GetStepSequenceRequest) returns (GetStepSequenceResponse);

7
tensorflow/go/saved_model.go
View File

@ -21,8 +21,7 @@ import (
"unsafe"
"github.com/golang/protobuf/proto"
tfpb "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf"
corepb "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto"
)
// #include <stdlib.h>
@ -73,7 +72,7 @@ func LoadSavedModel(exportDir string, tags []string, options *SessionOptions) (*
C.free(unsafe.Pointer(cExportDir))
metaGraphDefBytes := C.GoBytes(metaGraphDefBuf.data, C.int(metaGraphDefBuf.length))
metaGraphDef := new(tfpb.MetaGraphDef)
metaGraphDef := new(corepb.MetaGraphDef)
if err := proto.Unmarshal(metaGraphDefBytes, metaGraphDef); err != nil {
return nil, err
}
@ -88,7 +87,7 @@ func LoadSavedModel(exportDir string, tags []string, options *SessionOptions) (*
return &SavedModel{Session: s, Graph: graph, Signatures: signatures}, nil
}
func generateSignatures(pb map[string]*tfpb.SignatureDef) map[string]Signature {
func generateSignatures(pb map[string]*corepb.SignatureDef) map[string]Signature {
signatures := make(map[string]Signature)
for name, signature := range pb {
signatures[name] = signatureDefFromProto(signature)

6
tensorflow/go/signature.go
View File

@ -16,7 +16,7 @@ limitations under the License.
package tensorflow
import tfpb "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf"
import corepb "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto"
// #include "tensorflow/c/c_api.h"
import "C"
@ -90,7 +90,7 @@ type TensorInfo struct {
Shape Shape
}
func signatureDefFromProto(pb *tfpb.SignatureDef) Signature {
func signatureDefFromProto(pb *corepb.SignatureDef) Signature {
inputs := make(map[string]TensorInfo)
for name, input := range pb.GetInputs() {
inputs[name] = tensorInfoFromProto(input)
@ -106,7 +106,7 @@ func signatureDefFromProto(pb *tfpb.SignatureDef) Signature {
}
}
func tensorInfoFromProto(pb *tfpb.TensorInfo) TensorInfo {
func tensorInfoFromProto(pb *corepb.TensorInfo) TensorInfo {
var dims []int64
for _, d := range pb.GetTensorShape().GetDim() {
dims = append(dims, d.GetSize())

28
tensorflow/go/signature_test.go
View File

@ -20,16 +20,16 @@ import (
"fmt"
"testing"
corepb "github.com/tensorflow/tensorflow/tensorflow/go/core/core_protos_go_proto"
tspb "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_shape_go_proto"
typb "github.com/tensorflow/tensorflow/tensorflow/go/core/framework/types_go_proto"
tfpb "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf"
)
func TestSignatureFromProto(t *testing.T) {
got := signatureDefFromProto(&tfpb.SignatureDef{
Inputs: map[string]*tfpb.TensorInfo{
"input_1": &tfpb.TensorInfo{
Encoding: &tfpb.TensorInfo_Name{
got := signatureDefFromProto(&corepb.SignatureDef{
Inputs: map[string]*corepb.TensorInfo{
"input_1": &corepb.TensorInfo{
Encoding: &corepb.TensorInfo_Name{
Name: "tensor_1",
},
Dtype: typb.DataType_DT_INT8,
@ -41,8 +41,8 @@ func TestSignatureFromProto(t *testing.T) {
},
},
},
"input_2": &tfpb.TensorInfo{
Encoding: &tfpb.TensorInfo_Name{
"input_2": &corepb.TensorInfo{
Encoding: &corepb.TensorInfo_Name{
Name: "tensor_2",
},
Dtype: typb.DataType_DT_FLOAT,
@ -55,9 +55,9 @@ func TestSignatureFromProto(t *testing.T) {
},
},
},
Outputs: map[string]*tfpb.TensorInfo{
"output_1": &tfpb.TensorInfo{
Encoding: &tfpb.TensorInfo_Name{
Outputs: map[string]*corepb.TensorInfo{
"output_1": &corepb.TensorInfo{
Encoding: &corepb.TensorInfo_Name{
Name: "tensor_3",
},
Dtype: typb.DataType_DT_STRING,
@ -69,8 +69,8 @@ func TestSignatureFromProto(t *testing.T) {
},
},
},
"output_2": &tfpb.TensorInfo{
Encoding: &tfpb.TensorInfo_Name{
"output_2": &corepb.TensorInfo{
Encoding: &corepb.TensorInfo_Name{
Name: "tensor_4",
},
Dtype: typb.DataType_DT_BOOL,
@ -140,8 +140,8 @@ func TestSignatureFromProto(t *testing.T) {
}
func TestTensorInfoFromProto(t *testing.T) {
got := tensorInfoFromProto(&tfpb.TensorInfo{
Encoding: &tfpb.TensorInfo_Name{
got := tensorInfoFromProto(&corepb.TensorInfo{
Encoding: &corepb.TensorInfo_Name{
Name: "tensor",
},
Dtype: typb.DataType_DT_INT8,