Home
last modified time | relevance | path

Searched refs:debug_node_key (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/debug/
Ddebug_grpc_testlib.cc82 const DebugNodeKey debug_node_key(metadata.device(), node_name, in SendEvents() local
84 if (write_enabled_debug_node_keys_.find(debug_node_key) != in SendEvents()
101 const DebugNodeKey& debug_node_key = debug_node_keys_[i]; in SendEvents() local
102 change->set_node_name(debug_node_key.node_name); in SendEvents()
103 change->set_output_slot(debug_node_key.output_slot); in SendEvents()
104 change->set_debug_op(debug_node_key.debug_op); in SendEvents()
108 write_enabled_debug_node_keys_.insert(debug_node_key); in SendEvents()
110 write_enabled_debug_node_keys_.erase(debug_node_key); in SendEvents()
133 const DebugNodeKey& debug_node_key) { in RequestDebugOpStateChangeAtNextStream() argument
136 debug_node_keys_.push_back(debug_node_key); in RequestDebugOpStateChangeAtNextStream()
Ddebug_io_utils.cc61 Event PrepareChunkEventProto(const DebugNodeKey& debug_node_key, in PrepareChunkEventProto() argument
74 value->set_node_name(debug_node_key.debug_node_name); in PrepareChunkEventProto()
78 value->set_tag(debug_node_key.node_name); in PrepareChunkEventProto()
82 metadata.set_device(debug_node_key.device_name); in PrepareChunkEventProto()
83 metadata.set_output_slot(debug_node_key.output_slot); in PrepareChunkEventProto()
101 << "The debug_node_name is " << debug_node_key.debug_node_name in PrepareChunkEventProto()
126 Status WrapStringTensorAsEvents(const DebugNodeKey& debug_node_key, in WrapStringTensorAsEvents() argument
146 debug_node_key.debug_node_name, in WrapStringTensorAsEvents()
159 Event event = PrepareChunkEventProto(debug_node_key, wall_time_us, in WrapStringTensorAsEvents()
186 Status WrapTensorAsEvents(const DebugNodeKey& debug_node_key, in WrapTensorAsEvents() argument
[all …]
Ddebug_io_utils.h78 static Status PublishDebugTensor(const DebugNodeKey& debug_node_key,
85 static Status PublishDebugTensor(const DebugNodeKey& debug_node_key,
168 static Status DumpTensorToDir(const DebugNodeKey& debug_node_key,
183 const DebugNodeKey& debug_node_key,
218 static Status DumpTensorToEventFile(const DebugNodeKey& debug_node_key,
341 static Status SendTensorThroughGrpcStream(const DebugNodeKey& debug_node_key,
DBUILD13 # ":debug_node_key" - Defines a struct used for tracking tensors.
124 ":debug_node_key",
160 name = "debug_node_key",
161 srcs = ["debug_node_key.cc"],
162 hdrs = ["debug_node_key.h"],
180 ":debug_node_key",
274 ":debug_node_key",
Ddebug_io_utils_test.cc60 DebugNodeKey debug_node_key("/job:worker/replica:1/task:0/device:GPU:2", in TEST_F() local
63 debug_node_key.device_name); in TEST_F()
64 EXPECT_EQ("hidden_1/MatMul", debug_node_key.node_name); in TEST_F()
65 EXPECT_EQ(0, debug_node_key.output_slot); in TEST_F()
66 EXPECT_EQ("DebugIdentity", debug_node_key.debug_op); in TEST_F()
67 EXPECT_EQ("hidden_1/MatMul:0:DebugIdentity", debug_node_key.debug_node_name); in TEST_F()
69 debug_node_key.device_path); in TEST_F()
Ddebug_grpc_testlib.h49 const DebugNodeKey& debug_node_key);
/external/tensorflow/tensorflow/python/debug/lib/
Dgrpc_debug_server.py201 debug_node_key = (state_change.node_name, state_change.output_slot,
207 self._breakpoints.add(debug_node_key)
212 if debug_node_key in self._breakpoints:
213 self._breakpoints.discard(debug_node_key)
219 if debug_node_key in self._breakpoints:
220 self._breakpoints.discard(debug_node_key)
224 debug_node_key)
/external/tensorflow/tensorflow/core/kernels/
DBUILD1722 "//tensorflow/core/debug:debug_node_key",