xref: /aosp_15_r20/external/federated-compute/fcp/tensorflow/test_selector.proto (revision 14675a029014e728ec732f129a32e299b2da0601)
1/*
2 * Copyright 2019 Google LLC
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17// Defines TestSelector, TestCriteria and ResumptionToken, for use in
18// ExternalDataset and ExampleSelectorFuser testing.
19
20syntax = "proto3";
21
22package fcp;
23
24message Limit {
25  int64 value = 1;
26}
27
28// Specifies that examples (int64 scalars) in the range [lower_inclusive,
29// upper_inclusive] should be included.
30message TestSelector {
31  Limit lower_inclusive = 1;
32  Limit upper_inclusive = 2;
33}
34
35// Simple example selection criteria which limits the maximum number of examples
36message TestCriteria {
37  // The max number of examples that should be returned by this query.
38  int32 max_examples = 1;
39}
40
41message ResumptionToken {
42  int32 last_index = 1;
43}
44