• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1/* Copyright 2016 The TensorFlow Authors. All Rights Reserved.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7    http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is distributed on an "AS IS" BASIS,
11WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14==============================================================================*/
15
16syntax = "proto3";
17
18package tensorflow;
19
20import "tensorflow/core/framework/tensor.proto";
21import "tensorflow/core/profiler/tfprof_log.proto";
22import "tensorflow/core/protobuf/debug.proto";
23import "tensorflow/core/util/event.proto";
24
25// Reply message from EventListener to the client, i.e., to the source of the
26// Event protocol buffers, e.g., debug ops inserted by a debugged runtime to a
27// TensorFlow graph being executed.
28message EventReply {
29  message DebugOpStateChange {
30    enum State {
31      STATE_UNSPECIFIED = 0;
32      DISABLED = 1;
33      READ_ONLY = 2;
34      READ_WRITE = 3;
35    }
36
37    State state = 1;
38    string node_name = 2;
39    int32 output_slot = 3;
40    string debug_op = 4;
41  }
42
43  repeated DebugOpStateChange debug_op_state_changes = 1;
44
45  // New tensor value to override the current tensor value with.
46  TensorProto tensor = 2;
47  // TODO(cais): Make use of this field to implement overriding of tensor value
48  // during debugging.
49}
50
51// Data on the traceback of a debugged call, e.g., a Session.run() call, or the
52// execution of an eager operation.
53message CallTraceback {
54  enum CallType {
55    UNSPECIFIED = 0;
56    GRAPH_EXECUTION = 1;
57    EAGER_EXECUTION = 2;
58  }
59
60  CallType call_type = 1;
61
62  // A key for the call. For example, for graph execution, this is a key
63  // consisting of the names of the fed and fetched tensors.
64  string call_key = 2;
65
66  // Traceback stack for the origin of the call event.
67  // For graph execution, this is the stack of the Session.run() call.
68  // For eager execution, this is the stack of the Python line that invokes
69  // the execution of the eager op.
70  tfprof.CodeDef origin_stack = 3;
71
72  // Keeps track of the mapping from integer IDs in `origin_stack` to actual
73  // string values (e.g., file paths, function names).
74  map<int64, string> origin_id_to_string = 4;
75
76  // Traceback for the graph (if any) involved in the call.
77  tfprof.OpLogProto graph_traceback = 5;
78
79  // Version of the graph in `graph_traceback` (if any).
80  int64 graph_version = 6;
81}
82
83// EventListener: Receives Event protos, e.g., from debugged TensorFlow
84// runtime(s).
85service EventListener {
86  // Client(s) can use this RPC method to send the EventListener Event protos.
87  // The Event protos can hold information such as:
88  //   1) intermediate tensors from a debugged graph being executed, which can
89  //      be sent from DebugIdentity ops configured with grpc URLs.
90  //   2) GraphDefs of partition graphs, which can be sent from special debug
91  //      ops that get executed immediately after the beginning of the graph
92  //      execution.
93  rpc SendEvents(stream Event) returns (stream EventReply);
94
95  // Send the tracebacks of a TensorFlow execution call.
96  rpc SendTracebacks(CallTraceback) returns (EventReply);
97
98  // Send a collection of source code files being debugged.
99  rpc SendSourceFiles(DebuggedSourceFiles) returns (EventReply);
100}
101