HomeSort by relevance Sort by last modified time
    Searched refs:debug_node_key (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/debug/
debug_grpc_testlib.cc 82 const DebugNodeKey debug_node_key(metadata.device(), node_name,
84 if (write_enabled_debug_node_keys_.find(debug_node_key) !=
101 const DebugNodeKey& debug_node_key = debug_node_keys_[i]; local
102 change->set_node_name(debug_node_key.node_name);
103 change->set_output_slot(debug_node_key.output_slot);
104 change->set_debug_op(debug_node_key.debug_op);
108 write_enabled_debug_node_keys_.insert(debug_node_key);
110 write_enabled_debug_node_keys_.erase(debug_node_key);
133 const DebugNodeKey& debug_node_key) {
136 debug_node_keys_.push_back(debug_node_key);
    [all...]
debug_io_utils.cc 61 Event PrepareChunkEventProto(const DebugNodeKey& debug_node_key,
74 value->set_node_name(debug_node_key.debug_node_name);
78 value->set_tag(debug_node_key.node_name);
82 metadata.set_device(debug_node_key.device_name);
83 metadata.set_output_slot(debug_node_key.output_slot);
101 << "The debug_node_name is " << debug_node_key.debug_node_name
126 Status WrapStringTensorAsEvents(const DebugNodeKey& debug_node_key,
146 debug_node_key.debug_node_name,
159 Event event = PrepareChunkEventProto(debug_node_key, wall_time_us,
186 Status WrapTensorAsEvents(const DebugNodeKey& debug_node_key,
    [all...]
debug_io_utils.h 27 #include "tensorflow/core/debug/debug_node_key.h"
72 // debug_node_key: A DebugNodeKey identifying the debug node.
78 static Status PublishDebugTensor(const DebugNodeKey& debug_node_key,
85 static Status PublishDebugTensor(const DebugNodeKey& debug_node_key,
163 // debug_node_key: A DebugNodeKey identifying the debug node.
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,
debug_grpc_testlib.h 49 const DebugNodeKey& debug_node_key);
debug_io_utils_test.cc 22 #include "tensorflow/core/debug/debug_node_key.h"
60 DebugNodeKey debug_node_key("/job:worker/replica:1/task:0/device:GPU:2",
63 debug_node_key.device_name);
64 EXPECT_EQ("hidden_1/MatMul", debug_node_key.node_name);
65 EXPECT_EQ(0, debug_node_key.output_slot);
66 EXPECT_EQ("DebugIdentity", debug_node_key.debug_op);
67 EXPECT_EQ("hidden_1/MatMul:0:DebugIdentity", debug_node_key.debug_node_name);
69 debug_node_key.device_path);
  /external/tensorflow/tensorflow/python/debug/lib/
grpc_debug_server.py 201 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)

Completed in 53 milliseconds