Remove TensorFlow dependency from the federating_executor module.

PiperOrigin-RevId: 746084493

Author: michaelreneer

tensorflow_federated/cc/core/impl/executors/BUILD

@@ -453,7 +453,6 @@ cc_library(
         ":executor",
         ":federated_intrinsics",
         ":status_macros",
-        ":tensor_serialization",
         ":threading",
         ":value_validation",
         "//tensorflow_federated/proto/v0:executor_cc_proto",
@@ -466,29 +465,30 @@ cc_library(
         "@com_google_absl//absl/strings:string_view",
         "@com_google_absl//absl/synchronization",
         "@com_google_absl//absl/types:span",
+        "@federated_language//federated_language/proto:array_cc_proto",
         "@federated_language//federated_language/proto:computation_cc_proto",
-        "@org_tensorflow//tensorflow/core:framework",
+        "@federated_language//federated_language/proto:data_type_cc_proto",
     ],
 )
 
 cc_test(
     name = "federating_executor_test",
     srcs = ["federating_executor_test.cc"],
     deps = [
+        ":array_shape_test_utils",
+        ":array_test_utils",
         ":executor",
         ":executor_test_base",
         ":federating_executor",
         ":mock_executor",
         ":status_macros",
-        ":tensorflow_test_utils",
         ":value_test_utils",
         "//tensorflow_federated/cc/testing:oss_test_main",
         "//tensorflow_federated/cc/testing:status_matchers",
         "//tensorflow_federated/proto/v0:executor_cc_proto",
         "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/types:span",
-        "@org_tensorflow//tensorflow/core:framework",
         "@org_tensorflow//tensorflow/core:tensorflow",
     ],
 )
@@ -1000,7 +1000,7 @@ cc_library(
     deps = [
         ":dataset_utils",
         ":status_macros",
-        ":tensor_serialization",
+        ":tensorflow_utils",
         "//tensorflow_federated/cc/testing:protobuf_matchers",
         "//tensorflow_federated/proto/v0:executor_cc_proto",
         "@com_google_absl//absl/status",

tensorflow_federated/cc/core/impl/executors/federating_executor.cc

@@ -33,13 +33,13 @@ limitations under the License
 #include "absl/strings/string_view.h"
 #include "absl/synchronization/mutex.h"
 #include "absl/types/span.h"
+#include "federated_language/proto/array.pb.h"
 #include "federated_language/proto/computation.pb.h"
-#include "tensorflow/core/framework/tensor.h"
+#include "federated_language/proto/data_type.pb.h"
 #include "tensorflow_federated/cc/core/impl/executors/cardinalities.h"
 #include "tensorflow_federated/cc/core/impl/executors/executor.h"
 #include "tensorflow_federated/cc/core/impl/executors/federated_intrinsics.h"
 #include "tensorflow_federated/cc/core/impl/executors/status_macros.h"
-#include "tensorflow_federated/cc/core/impl/executors/tensor_serialization.h"
 #include "tensorflow_federated/cc/core/impl/executors/threading.h"
 #include "tensorflow_federated/cc/core/impl/executors/value_validation.h"
 #include "tensorflow_federated/proto/v0/executor.pb.h"
@@ -703,24 +703,23 @@ class FederatingExecutor : public ExecutorBase<ExecutorValue> {
       // these materialize calls don't block.
       v0::Value keys_for_client_pb =
           TFF_TRY(client_child_->Materialize(keys_child_id->ref()));
-      tensorflow::Tensor keys_for_client_tensor =
-          TFF_TRY(DeserializeTensorValue(keys_for_client_pb));
-      if (keys_for_client_tensor.dtype() != tensorflow::DT_INT32) {
+      federated_language::Array array_pb = keys_for_client_pb.array();
+      if (array_pb.dtype() != federated_language::DataType::DT_INT32) {
         return absl::InvalidArgumentError(
             absl::StrCat("Expected int32_t key, found key of tensor dtype ",
-                         keys_for_client_tensor.dtype()));
+                         array_pb.dtype()));
       }
-      if (keys_for_client_tensor.dims() != 1) {
+      if (array_pb.shape().dim().size() != 1) {
         return absl::InvalidArgumentError(absl::StrCat(
             "Expected key tensor to be rank one, but found tensor of rank ",
-            keys_for_client_tensor.dims()));
+            array_pb.shape().dim().size()));
       }
-      int64_t num_keys = keys_for_client_tensor.NumElements();
+      int64_t num_keys = array_pb.shape().dim()[0];
       std::vector<int32_t> keys_for_client;
       keys_for_client.reserve(num_keys);
-      auto keys_for_client_eigen = keys_for_client_tensor.flat<int32_t>();
+      auto keys_for_client_eigen = array_pb.int32_list().value();
       for (int64_t i = 0; i < num_keys; i++) {
-        int32_t key = keys_for_client_eigen(i);
+        int32_t key = keys_for_client_eigen[i];
         keys_for_client.push_back(key);
         keys.all.insert(key);
       }
@@ -732,8 +731,12 @@ class FederatingExecutor : public ExecutorBase<ExecutorValue> {
   absl::StatusOr<OwnedValueId> SelectSliceForKey(int32_t key,
                                                  ValueId server_val_child_id,
                                                  ValueId select_fn_child_id) {
+    federated_language::Array array_pb;
+    array_pb.set_dtype(federated_language::DataType::DT_INT32);
+    array_pb.mutable_shape();
+    array_pb.mutable_int32_list()->add_value(key);
     v0::Value key_pb;
-    TFF_TRY(SerializeTensorValue(tensorflow::Tensor(key), &key_pb));
+    *key_pb.mutable_array() = array_pb;
     OwnedValueId key_id = TFF_TRY(server_child_->CreateValue(key_pb));
     OwnedValueId arg_id =
         TFF_TRY(server_child_->CreateStruct({server_val_child_id, key_id}));