xref: /aosp_15_r20/external/tensorflow/tensorflow/core/data/service/export.proto (revision b6fb3261f9314811a0f4371741dbb8839866f948) 
1syntax = "proto3";
2
3package tensorflow.data;
4
5import "tensorflow/core/data/service/common.proto";
6import "tensorflow/core/protobuf/data_service.proto";
7import "tensorflow/core/protobuf/service_config.proto";
8
9// State of the dispatcher server, exported to improve debuggability.
10// Next tag: 4
11message DispatcherStateExport {
12  message Iteration {
13    string dataset_id = 1;
14    int64 iteration_id = 2;
15    IterationKeyDef iteration_key = 3;
16    ProcessingModeDef processing_mode = 4;
17    int64 num_consumers = 6;
18    int64 num_clients = 8;
19    bool finished = 10;
20    bool garbage_collected = 11;
21  }
22
23  experimental.DispatcherConfig dispatcher_config = 1;
24  repeated string worker_addresses = 2;
25  repeated Iteration iterations = 3;
26}
27
28// State of the worker server, exported to improve debuggability.
29// Next tag: 5
30message WorkerStateExport {
31  experimental.WorkerConfig worker_config = 1;
32  repeated TaskDef tasks = 2;
33  repeated int64 finished_task_ids = 3;
34  repeated int64 deleted_task_ids = 4;
35}
36
37// State of the tf.data service server, exported to improve debuggability.
38// The dispatcher and worker servers will populate the corresponding fields.
39// Next tag: 3
40message ServerStateExport {
41  DispatcherStateExport dispatcher_state_export = 1;
42  WorkerStateExport worker_state_export = 2;
43}
44