xref: /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/replay_log.proto (revision b6fb3261f9314811a0f4371741dbb8839866f948)
1syntax = "proto3";
2
3package tensorflow;
4
5import "tensorflow/core/protobuf/master.proto";
6
7option cc_enable_arenas = true;
8option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto";
9
10// Records the creation of a new replay session.  We record the device listing
11// here to capture the state of the cluster.
12message NewReplaySession {
13  ListDevicesResponse devices = 1;
14  string session_handle = 2;
15}
16
17message ReplayOp {
18  double start_time_us = 31;
19  double end_time_us = 32;
20
21  oneof op {
22    CreateSessionRequest create_session = 1;
23    ExtendSessionRequest extend_session = 2;
24    PartialRunSetupRequest partial_run_setup = 3;
25    RunStepRequest run_step = 4;
26    CloseSessionRequest close_session = 5;
27    ListDevicesRequest list_devices = 6;
28    ResetRequest reset_request = 7;
29    MakeCallableRequest make_callable = 8;
30    RunCallableRequest run_callable = 9;
31    ReleaseCallableRequest release_callable = 10;
32    NewReplaySession new_replay_session = 11;
33  }
34
35  oneof response {
36    CreateSessionResponse create_session_response = 21;
37    ExtendSessionResponse extend_session_response = 22;
38    PartialRunSetupResponse partial_run_setup_response = 23;
39    RunStepResponse run_step_response = 24;
40    CloseSessionResponse close_session_response = 25;
41    ListDevicesResponse list_devices_response = 26;
42    ResetResponse reset_request_response = 27;
43    MakeCallableResponse make_callable_response = 28;
44    RunCallableResponse run_callable_response = 29;
45    ReleaseCallableResponse release_callable_response = 30;
46  }
47}
48