Skip to content

Commit 688fcbb

Browse files
karllessardkarllessard
committed
Classify TF2.5 ops and protos
1 parent a406253 commit 688fcbb

File tree

168 files changed

+22994
-22450
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

168 files changed

+22994
-22450
lines changed

‎tensorflow-core/tensorflow-core-api/external/tensorflow-proto.patch

Lines changed: 66 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,17 @@ diff -ruN tensorflow-2.5.0-rc1/tensorflow/core/protobuf/trackable_object_graph.p
7676

7777
// A TensorBundle addition which saves extra information about the objects which
7878
// own variables, allowing for more robust checkpoint loading into modified
79+
diff -ruN tensorflow-2.5.0-rc1/tensorflow/core/protobuf/transport_options.proto tensorflow-2.5.0-rc1-proto/tensorflow/core/protobuf/transport_options.proto
80+
--- tensorflow-2.5.0-rc1/tensorflow/core/protobuf/transport_options.proto 2021-01-21 09:25:54.000000000 +0900
81+
+++ tensorflow-2.5.0-rc1-proto/tensorflow/core/protobuf/transport_options.proto 2021-02-08 09:42:56.660650580 +0900
82+
@@ -3,6 +3,7 @@
83+
package tensorflow;
84+
85+
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto";
86+
+option java_package = "org.tensorflow.distruntime";
87+
88+
// Extra data needed on a non-RDMA RecvBufResponse.
89+
message RecvBufRespExtra {
7990
diff -ruN tensorflow-2.5.0-rc1/tensorflow/core/lib/core/error_codes.proto tensorflow-2.5.0-rc1-proto/tensorflow/core/lib/core/error_codes.proto
8091
--- tensorflow-2.5.0-rc1/tensorflow/core/lib/core/error_codes.proto 2021-04-13 01:43:40.000000000 +0900
8192
+++ tensorflow-2.5.0-rc1-proto/tensorflow/core/lib/core/error_codes.proto 2021-04-27 10:19:52.003448629 +0900
@@ -135,3 +146,58 @@ diff -ruN tensorflow-2.5.0-rc1/tensorflow/core/protobuf/service_config.proto ten
135146

136147
// Configuration for a tf.data service DispatchServer.
137148
message DispatcherConfig {
149+
diff --git a/tensorflow/core/framework/dataset_options.proto b/tensorflow/core/framework/dataset_options.proto
150+
index 3d71a560956..4c427640148 100644
151+
--- a/tensorflow/core/framework/dataset_options.proto
152+
+++ b/tensorflow/core/framework/dataset_options.proto
153+
@@ -2,6 +2,10 @@ syntax = "proto3";
154+
155+
package tensorflow.data;
156+
157+
+option java_outer_classname = "DatasetOptionsProtos";
158+
+option java_multiple_files = true;
159+
+option java_package = "org.tensorflow.data";
160+
+
161+
// Represents the type of auto-sharding we enable.
162+
enum AutoShardPolicy {
163+
// AUTO: Attempts FILE-based sharding, falling back to DATA-based sharding.
164+
diff --git a/tensorflow/core/framework/model.proto b/tensorflow/core/framework/model.proto
165+
index ba74d7a2b7e..721dee57867 100644
166+
--- a/tensorflow/core/framework/model.proto
167+
+++ b/tensorflow/core/framework/model.proto
168+
@@ -3,6 +3,9 @@ syntax = "proto3";
169+
package tensorflow.data.model;
170+
171+
option cc_enable_arenas = true;
172+
+option java_outer_classname = "ModelProtos";
173+
+option java_multiple_files = true;
174+
+option java_package = "org.tensorflow.data.model";
175+
176+
// Class of a node in the performance model.
177+
enum NodeClass {
178+
diff --git a/tensorflow/core/grappler/costs/op_performance_data.proto b/tensorflow/core/grappler/costs/op_performance_data.proto
179+
index 5ef5fd927b8..7c9a6ca2141 100644
180+
--- a/tensorflow/core/grappler/costs/op_performance_data.proto
181+
+++ b/tensorflow/core/grappler/costs/op_performance_data.proto
182+
@@ -17,6 +17,9 @@ syntax = "proto3";
183+
184+
package tensorflow;
185+
option cc_enable_arenas = true;
186+
+option java_outer_classname = "OpPerformanceDataProtos";
187+
+option java_multiple_files = true;
188+
+option java_package = "org.tensorflow.framework";
189+
190+
import "tensorflow/core/framework/tensor.proto";
191+
import "tensorflow/core/framework/tensor_shape.proto";
192+
diff --git a/tensorflow/core/protobuf/extension_type_variant.proto b/tensorflow/core/protobuf/extension_type_variant.proto
193+
index 536db3b2435..88c4701b505 100644
194+
--- a/tensorflow/core/protobuf/extension_type_variant.proto
195+
+++ b/tensorflow/core/protobuf/extension_type_variant.proto
196+
@@ -3,6 +3,7 @@ syntax = "proto3";
197+
package tensorflow;
198+
199+
import "tensorflow/core/protobuf/struct.proto";
200+
+option java_package = "org.tensorflow.framework";
201+
202+
// Metadata for ExtensionTypeVariant, used when serializing as Variant.
203+
//

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CollectiveBcastRecvV2.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "CollectiveBcastRecvV2"
3+
endpoint {
4+
name: "rawops.CollectiveBcastRecvV2"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_CollectiveBcastSendV2.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "CollectiveBcastSendV2"
3+
endpoint {
4+
name: "rawops.CollectiveBcastSendV2"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_DataServiceDatasetV2.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "DataServiceDatasetV2"
3+
endpoint {
4+
name: "rawops.DataServiceDatasetV2"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_FinalizeDataset.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "FinalizeDataset"
3+
endpoint {
4+
name: "rawops.FinalizeDataset"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_GetOptions.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "GetOptions"
3+
endpoint {
4+
name: "rawops.GetOptions"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_LoadTPUEmbeddingFrequencyEstimatorParameters.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "LoadTPUEmbeddingFrequencyEstimatorParameters"
3+
endpoint {
4+
name: "rawops.LoadTPUEmbeddingFrequencyEstimatorParameters"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug"
3+
endpoint {
4+
name: "rawops.LoadTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_OptionsDataset.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "OptionsDataset"
3+
endpoint {
4+
name: "rawops.OptionsDataset"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_ParallelBatchDataset.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "ParallelBatchDataset"
3+
endpoint {
4+
name: "rawops.ParallelBatchDataset"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParameters.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RetrieveTPUEmbeddingFrequencyEstimatorParameters"
3+
endpoint {
4+
name: "rawops.RetrieveTPUEmbeddingFrequencyEstimatorParameters"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug"
3+
endpoint {
4+
name: "rawops.RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebug"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscAbs.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscAbs"
3+
endpoint {
4+
name: "risc.RiscAbs"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscAdd.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscAdd"
3+
endpoint {
4+
name: "risc.RiscAdd"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscBinaryArithmetic.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscBinaryArithmetic"
3+
endpoint {
4+
name: "risc.RiscBinaryArithmetic"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscBinaryComparison.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscBinaryComparison"
3+
endpoint {
4+
name: "risc.RiscBinaryComparison"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscBitcast.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscBitcast"
3+
endpoint {
4+
name: "risc.RiscBitcast"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscBroadcast.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscBroadcast"
3+
endpoint {
4+
name: "risc.RiscBroadcast"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscCast.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscCast"
3+
endpoint {
4+
name: "risc.RiscCast"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscCeil.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscCeil"
3+
endpoint {
4+
name: "risc.RiscCeil"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscCholesky.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscCholesky"
3+
endpoint {
4+
name: "risc.RiscCholesky"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscConcat.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscConcat"
3+
endpoint {
4+
name: "risc.RiscConcat"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscCondition.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscCondition"
3+
endpoint {
4+
name: "risc.RiscCondition"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscConv.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscConv"
3+
endpoint {
4+
name: "risc.RiscConv"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscCos.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscCos"
3+
endpoint {
4+
name: "risc.RiscCos"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscDiv.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscDiv"
3+
endpoint {
4+
name: "risc.RiscDiv"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscDot.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscDot"
3+
endpoint {
4+
name: "risc.RiscDot"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscExp.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscExp"
3+
endpoint {
4+
name: "risc.RiscExp"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscFft.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscFft"
3+
endpoint {
4+
name: "risc.RiscFft"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscFloor.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscFloor"
3+
endpoint {
4+
name: "risc.RiscFloor"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscGather.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscGather"
3+
endpoint {
4+
name: "risc.RiscGather"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscImag.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscImag"
3+
endpoint {
4+
name: "risc.RiscImag"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscIsFinite.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscIsFinite"
3+
endpoint {
4+
name: "risc.RiscIsFinite"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscLog.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscLog"
3+
endpoint {
4+
name: "risc.RiscLog"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscLogicalAnd.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscLogicalAnd"
3+
endpoint {
4+
name: "risc.RiscLogicalAnd"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscLogicalNot.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscLogicalNot"
3+
endpoint {
4+
name: "risc.RiscLogicalNot"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscLogicalOr.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscLogicalOr"
3+
endpoint {
4+
name: "risc.RiscLogicalOr"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscMax.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscMax"
3+
endpoint {
4+
name: "risc.RiscMax"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscMin.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscMin"
3+
endpoint {
4+
name: "risc.RiscMin"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscMul.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscMul"
3+
endpoint {
4+
name: "risc.RiscMul"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscNeg.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscNeg"
3+
endpoint {
4+
name: "risc.RiscNeg"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscPad.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscPad"
3+
endpoint {
4+
name: "risc.RiscPad"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscPool.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscPool"
3+
endpoint {
4+
name: "risc.RiscPool"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscPow.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscPow"
3+
endpoint {
4+
name: "risc.RiscPow"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscRandomUniform.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscRandomUniform"
3+
endpoint {
4+
name: "risc.RiscRandomUniform"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscReal.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscReal"
3+
endpoint {
4+
name: "risc.RiscReal"
5+
}
6+
}

‎tensorflow-core/tensorflow-core-api/src/bazel/api_def/api_def_RiscReduce.pbtxt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
op {
2+
graph_op_name: "RiscReduce"
3+
endpoint {
4+
name: "risc.RiscReduce"
5+
}
6+
}

0 commit comments

Comments
 (0)