Skip to content

Commit f3a51da

Browse files
committed
Update protobuf per tf 2.6.
1 parent 824dfe6 commit f3a51da

File tree

12 files changed

+1756
-295
lines changed

12 files changed

+1756
-295
lines changed

src/TensorFlowNET.Core/Protobuf/Config.cs

Lines changed: 287 additions & 106 deletions
Large diffs are not rendered by default.

src/TensorFlowNET.Core/Protobuf/CppShapeInference.cs

Lines changed: 43 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -25,27 +25,28 @@ static CppShapeInferenceReflection() {
2525
byte[] descriptorData = global::System.Convert.FromBase64String(
2626
string.Concat(
2727
"CjV0ZW5zb3JmbG93L3B5dGhvbi9mcmFtZXdvcmsvY3BwX3NoYXBlX2luZmVy",
28-
"ZW5jZS5wcm90bxIKdGVuc29yZmxvdxoldGVuc29yZmxvdy9jb3JlL2ZyYW1l",
29-
"d29yay90eXBlcy5wcm90bxosdGVuc29yZmxvdy9jb3JlL2ZyYW1ld29yay90",
30-
"ZW5zb3Jfc2hhcGUucHJvdG8ipQMKF0NwcFNoYXBlSW5mZXJlbmNlUmVzdWx0",
28+
"ZW5jZS5wcm90bxIKdGVuc29yZmxvdxopdGVuc29yZmxvdy9jb3JlL2ZyYW1l",
29+
"d29yay9mdWxsX3R5cGUucHJvdG8aLHRlbnNvcmZsb3cvY29yZS9mcmFtZXdv",
30+
"cmsvdGVuc29yX3NoYXBlLnByb3RvGiV0ZW5zb3JmbG93L2NvcmUvZnJhbWV3",
31+
"b3JrL3R5cGVzLnByb3RvIpsDChdDcHBTaGFwZUluZmVyZW5jZVJlc3VsdBIr",
32+
"CgVzaGFwZRgBIAEoCzIcLnRlbnNvcmZsb3cuVGVuc29yU2hhcGVQcm90bxJD",
33+
"CgtoYW5kbGVfZGF0YRgEIAEoCzIuLnRlbnNvcmZsb3cuQ3BwU2hhcGVJbmZl",
34+
"cmVuY2VSZXN1bHQuSGFuZGxlRGF0YRqTAQoSSGFuZGxlU2hhcGVBbmRUeXBl",
3135
"EisKBXNoYXBlGAEgASgLMhwudGVuc29yZmxvdy5UZW5zb3JTaGFwZVByb3Rv",
32-
"EkMKC2hhbmRsZV9kYXRhGAQgASgLMi4udGVuc29yZmxvdy5DcHBTaGFwZUlu",
33-
"ZmVyZW5jZVJlc3VsdC5IYW5kbGVEYXRhGp0BChJIYW5kbGVTaGFwZUFuZFR5",
34-
"cGUSKwoFc2hhcGUYASABKAsyHC50ZW5zb3JmbG93LlRlbnNvclNoYXBlUHJv",
35-
"dG8SIwoFZHR5cGUYAiABKA4yFC50ZW5zb3JmbG93LkRhdGFUeXBlEjUKEHNw",
36-
"ZWNpYWxpemVkX3R5cGUYAyABKA4yGy50ZW5zb3JmbG93LlNwZWNpYWxpemVk",
37-
"VHlwZRpsCgpIYW5kbGVEYXRhEg4KBmlzX3NldBgBIAEoCBJOCg5zaGFwZV9h",
38-
"bmRfdHlwZRgCIAMoCzI2LnRlbnNvcmZsb3cuQ3BwU2hhcGVJbmZlcmVuY2VS",
39-
"ZXN1bHQuSGFuZGxlU2hhcGVBbmRUeXBlSgQIAhADSgQIAxAEImUKHUNwcFNo",
40-
"YXBlSW5mZXJlbmNlSW5wdXRzTmVlZGVkEhwKFGlucHV0X3RlbnNvcnNfbmVl",
41-
"ZGVkGAEgAygFEiYKHmlucHV0X3RlbnNvcnNfYXNfc2hhcGVzX25lZWRlZBgC",
42-
"IAMoBUJhWlxnaXRodWIuY29tL3RlbnNvcmZsb3cvdGVuc29yZmxvdy90ZW5z",
43-
"b3JmbG93L2dvL3B5dGhvbi9mcmFtZXdvcmsvY3BwX3NoYXBlX2luZmVyZW5j",
44-
"ZV9nb19wcm90b/gBAWIGcHJvdG8z"));
36+
"EiMKBWR0eXBlGAIgASgOMhQudGVuc29yZmxvdy5EYXRhVHlwZRIlCgR0eXBl",
37+
"GAQgASgLMhcudGVuc29yZmxvdy5GdWxsVHlwZURlZkoECAMQBBpsCgpIYW5k",
38+
"bGVEYXRhEg4KBmlzX3NldBgBIAEoCBJOCg5zaGFwZV9hbmRfdHlwZRgCIAMo",
39+
"CzI2LnRlbnNvcmZsb3cuQ3BwU2hhcGVJbmZlcmVuY2VSZXN1bHQuSGFuZGxl",
40+
"U2hhcGVBbmRUeXBlSgQIAhADSgQIAxAEImUKHUNwcFNoYXBlSW5mZXJlbmNl",
41+
"SW5wdXRzTmVlZGVkEhwKFGlucHV0X3RlbnNvcnNfbmVlZGVkGAEgAygFEiYK",
42+
"HmlucHV0X3RlbnNvcnNfYXNfc2hhcGVzX25lZWRlZBgCIAMoBUJhWlxnaXRo",
43+
"dWIuY29tL3RlbnNvcmZsb3cvdGVuc29yZmxvdy90ZW5zb3JmbG93L2dvL3B5",
44+
"dGhvbi9mcmFtZXdvcmsvY3BwX3NoYXBlX2luZmVyZW5jZV9nb19wcm90b/gB",
45+
"AWIGcHJvdG8z"));
4546
descriptor = pbr::FileDescriptor.FromGeneratedCode(descriptorData,
46-
new pbr::FileDescriptor[] { global::Tensorflow.TypesReflection.Descriptor, global::Tensorflow.TensorShapeReflection.Descriptor, },
47+
new pbr::FileDescriptor[] { global::Tensorflow.FullTypeReflection.Descriptor, global::Tensorflow.TensorShapeReflection.Descriptor, global::Tensorflow.TypesReflection.Descriptor, },
4748
new pbr::GeneratedClrTypeInfo(null, null, new pbr::GeneratedClrTypeInfo[] {
48-
new pbr::GeneratedClrTypeInfo(typeof(global::Tensorflow.CppShapeInferenceResult), global::Tensorflow.CppShapeInferenceResult.Parser, new[]{ "Shape", "HandleData" }, null, null, null, new pbr::GeneratedClrTypeInfo[] { new pbr::GeneratedClrTypeInfo(typeof(global::Tensorflow.CppShapeInferenceResult.Types.HandleShapeAndType), global::Tensorflow.CppShapeInferenceResult.Types.HandleShapeAndType.Parser, new[]{ "Shape", "Dtype", "SpecializedType" }, null, null, null, null),
49+
new pbr::GeneratedClrTypeInfo(typeof(global::Tensorflow.CppShapeInferenceResult), global::Tensorflow.CppShapeInferenceResult.Parser, new[]{ "Shape", "HandleData" }, null, null, null, new pbr::GeneratedClrTypeInfo[] { new pbr::GeneratedClrTypeInfo(typeof(global::Tensorflow.CppShapeInferenceResult.Types.HandleShapeAndType), global::Tensorflow.CppShapeInferenceResult.Types.HandleShapeAndType.Parser, new[]{ "Shape", "Dtype", "Type" }, null, null, null, null),
4950
new pbr::GeneratedClrTypeInfo(typeof(global::Tensorflow.CppShapeInferenceResult.Types.HandleData), global::Tensorflow.CppShapeInferenceResult.Types.HandleData.Parser, new[]{ "IsSet", "ShapeAndType" }, null, null, null, null)}),
5051
new pbr::GeneratedClrTypeInfo(typeof(global::Tensorflow.CppShapeInferenceInputsNeeded), global::Tensorflow.CppShapeInferenceInputsNeeded.Parser, new[]{ "InputTensorsNeeded", "InputTensorsAsShapesNeeded" }, null, null, null, null)
5152
}));
@@ -252,7 +253,7 @@ public HandleShapeAndType() {
252253
public HandleShapeAndType(HandleShapeAndType other) : this() {
253254
shape_ = other.shape_ != null ? other.shape_.Clone() : null;
254255
dtype_ = other.dtype_;
255-
specializedType_ = other.specializedType_;
256+
type_ = other.type_ != null ? other.type_.Clone() : null;
256257
_unknownFields = pb::UnknownFieldSet.Clone(other._unknownFields);
257258
}
258259

@@ -283,19 +284,14 @@ public HandleShapeAndType Clone() {
283284
}
284285
}
285286

286-
/// <summary>Field number for the "specialized_type" field.</summary>
287-
public const int SpecializedTypeFieldNumber = 3;
288-
private global::Tensorflow.SpecializedType specializedType_ = global::Tensorflow.SpecializedType.StInvalid;
289-
/// <summary>
290-
/// For dtype==DT_VARIANT, specialized_type may indicate a more specific
291-
/// type. For other dtypes or when the information is unavailable it is set
292-
/// to ST_INVALID.
293-
/// </summary>
287+
/// <summary>Field number for the "type" field.</summary>
288+
public const int TypeFieldNumber = 4;
289+
private global::Tensorflow.FullTypeDef type_;
294290
[global::System.Diagnostics.DebuggerNonUserCodeAttribute]
295-
public global::Tensorflow.SpecializedType SpecializedType {
296-
get { return specializedType_; }
291+
public global::Tensorflow.FullTypeDef Type {
292+
get { return type_; }
297293
set {
298-
specializedType_ = value;
294+
type_ = value;
299295
}
300296
}
301297

@@ -314,7 +310,7 @@ public bool Equals(HandleShapeAndType other) {
314310
}
315311
if (!object.Equals(Shape, other.Shape)) return false;
316312
if (Dtype != other.Dtype) return false;
317-
if (SpecializedType != other.SpecializedType) return false;
313+
if (!object.Equals(Type, other.Type)) return false;
318314
return Equals(_unknownFields, other._unknownFields);
319315
}
320316

@@ -323,7 +319,7 @@ public override int GetHashCode() {
323319
int hash = 1;
324320
if (shape_ != null) hash ^= Shape.GetHashCode();
325321
if (Dtype != global::Tensorflow.DataType.DtInvalid) hash ^= Dtype.GetHashCode();
326-
if (SpecializedType != global::Tensorflow.SpecializedType.StInvalid) hash ^= SpecializedType.GetHashCode();
322+
if (type_ != null) hash ^= Type.GetHashCode();
327323
if (_unknownFields != null) {
328324
hash ^= _unknownFields.GetHashCode();
329325
}
@@ -345,9 +341,9 @@ public void WriteTo(pb::CodedOutputStream output) {
345341
output.WriteRawTag(16);
346342
output.WriteEnum((int) Dtype);
347343
}
348-
if (SpecializedType != global::Tensorflow.SpecializedType.StInvalid) {
349-
output.WriteRawTag(24);
350-
output.WriteEnum((int) SpecializedType);
344+
if (type_ != null) {
345+
output.WriteRawTag(34);
346+
output.WriteMessage(Type);
351347
}
352348
if (_unknownFields != null) {
353349
_unknownFields.WriteTo(output);
@@ -363,8 +359,8 @@ public int CalculateSize() {
363359
if (Dtype != global::Tensorflow.DataType.DtInvalid) {
364360
size += 1 + pb::CodedOutputStream.ComputeEnumSize((int) Dtype);
365361
}
366-
if (SpecializedType != global::Tensorflow.SpecializedType.StInvalid) {
367-
size += 1 + pb::CodedOutputStream.ComputeEnumSize((int) SpecializedType);
362+
if (type_ != null) {
363+
size += 1 + pb::CodedOutputStream.ComputeMessageSize(Type);
368364
}
369365
if (_unknownFields != null) {
370366
size += _unknownFields.CalculateSize();
@@ -386,8 +382,11 @@ public void MergeFrom(HandleShapeAndType other) {
386382
if (other.Dtype != global::Tensorflow.DataType.DtInvalid) {
387383
Dtype = other.Dtype;
388384
}
389-
if (other.SpecializedType != global::Tensorflow.SpecializedType.StInvalid) {
390-
SpecializedType = other.SpecializedType;
385+
if (other.type_ != null) {
386+
if (type_ == null) {
387+
Type = new global::Tensorflow.FullTypeDef();
388+
}
389+
Type.MergeFrom(other.Type);
391390
}
392391
_unknownFields = pb::UnknownFieldSet.MergeFrom(_unknownFields, other._unknownFields);
393392
}
@@ -411,8 +410,11 @@ public void MergeFrom(pb::CodedInputStream input) {
411410
Dtype = (global::Tensorflow.DataType) input.ReadEnum();
412411
break;
413412
}
414-
case 24: {
415-
SpecializedType = (global::Tensorflow.SpecializedType) input.ReadEnum();
413+
case 34: {
414+
if (type_ == null) {
415+
Type = new global::Tensorflow.FullTypeDef();
416+
}
417+
input.ReadMessage(Type);
416418
break;
417419
}
418420
}

0 commit comments

Comments
 (0)