Add a configuration for a TFF Computation to be run in a Confidential…

… Compute

environment.

PiperOrigin-RevId: 644454277

fcp/protos/confidentialcompute/BUILD

@@ -183,6 +183,19 @@ java_lite_proto_library(
  deps = [":sql_query_proto"],
 )
 
+proto_library(
+ name = "tff_config_proto",
+ srcs = ["tff_config.proto"],
+ deps = [
+ "@org_tensorflow_federated//tensorflow_federated/proto/v0:executor_proto",
+ ],
+)
+
+cc_proto_library(
+ name = "tff_config_cc_proto",
+ deps = [":tff_config_proto"],
+)
+
 proto_library(
  name = "agg_core_container_config_proto",
  srcs = ["agg_core_container_config.proto"],

fcp/protos/confidentialcompute/pipeline_transform.proto

@@ -130,6 +130,9 @@ message Record {
 
  CompressionType compression_type = 3;
 
+ // Optional. Unique identifier for the record.
+ bytes record_id = 5;
+
  // Information about plaintext encrypted with AEAD, with the symmetric key
  // then encrypted with HPKE.
  message HpkePlusAeadData {

fcp/protos/confidentialcompute/tff_config.proto

@@ -0,0 +1,29 @@
+// Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+syntax = "proto3";
+
+package fcp.confidentialcompute;
+
+import "tensorflow_federated/proto/v0/executor.proto";
+
+option java_package = "com.google.intelligence.fcp.confidentialcompute";
+option java_multiple_files = true;
+
+// Configuration for a TFF Computation to be run in a Confidential Compute
+// environment.
+message TffConfig {
+ tensorflow_federated.v0.Value function = 1;
+ tensorflow_federated.v0.Value initial_arg = 2;
+}