| /external/tensorflow/tensorflow/core/debug/ |
| debug_node_key.cc | 27 const int32 output_slot, const string& debug_op)
30 output_slot(output_slot),
33 strings::StrCat(node_name, ":", output_slot, ":", debug_op)),
38 output_slot == other.output_slot && debug_op == other.debug_op);
|
| debug_gateway.cc | 28 [this](const string& node_name, const int output_slot,
30 if (comp_cb_ != nullptr && output_slot <= 0) {
33 // The output_slot can be negative (-1, or kControlSlot) if
37 comp_cb_(node_name, output_slot == 0);
42 if (val_cb_ != nullptr && output_slot >= 0) {
43 CopyTensor(node_name, output_slot, tensor, ctx,
44 [this, node_name, output_slot,
46 val_cb_(node_name, output_slot, *copied_tensor, is_ref);
68 void DebugGateway::CopyTensor(const string& node_name, const int output_slot,
80 string tensor_tag = strings::StrCat(node_name, ":", output_slot);
[all...] |
| debug_node_key.h | 31 const int32 output_slot, const string& debug_op);
43 const int32 output_slot; member in struct:tensorflow::DebugNodeKey
|
| debug_gateway.h | 52 // output_slot: Output slot number of the output Tensor
55 typedef std::function<void(const string& node_name, const int output_slot,
76 void CopyTensor(const string& node_name, const int output_slot,
|
| debug_service.proto | 39 int32 output_slot = 3;
|
| debug_grpc_testlib.cc | 74 output_slots.push_back(metadata.output_slot());
82 metadata.output_slot(), debug_op);
102 change->set_output_slot(debug_node_key.output_slot);
|
| debug_gateway_test.cc | 122 // output_slot values recorded in value callbacks
130 &callbacks_done](const string& node_name, const int output_slot,
135 output_slots_val.push_back(output_slot);
281 &callbacks_done](const string& node_name, const int output_slot,
286 } else if (node_name == debug_identity_node_name && output_slot == 0) {
287 // output_slot == 0 carries the debug signal. Same below.
289 } else if (node_name == debug_nan_count_node_name && output_slot == 0) {
394 &kConcurrentRuns](const string& node_name, const int output_slot,
398 if (node_name == a_debug_identity_node_name && output_slot == 0) {
402 output_slot == 0)
[all...] |
| debug_graph_utils.h | 86 const int output_slot);
|
| debug_io_utils_test.cc | 63 EXPECT_EQ(0, debug_node_key.output_slot);
176 ASSERT_EQ(kDebugNodeKey.output_slot, metadata.output_slot());
289 ASSERT_EQ(kDebugNodeKey.output_slot, metadata.output_slot());
434 ASSERT_EQ(kDebugNodeKey.output_slot, metadata.output_slot());
|
| debug_graph_utils.cc | 68 if (watch.output_slot() < 0) {
69 // The semantics of output_slot == -1 is that the node is watched only
79 strings::StrCat(watch.node_name(), ":", watch.output_slot());
248 const int output_slot) {
251 return strings::StrCat("__copy_", node_name, "_", output_slot);
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| evaluator_test.py | 31 device_name, node_name, output_slot, debug_op, exec_index = (
35 self.assertEqual(1, output_slot)
39 device_name, node_name, output_slot, debug_op, exec_index = (
43 self.assertEqual(0, output_slot)
48 device_name, node_name, output_slot, debug_op, exec_index = (
52 self.assertEqual(1, output_slot)
56 device_name, node_name, output_slot, debug_op, exec_index = (
61 self.assertEqual(0, output_slot)
66 device_name, node_name, output_slot, debug_op, exec_index = (
71 self.assertEqual(1, output_slot)
[all...] |
| evaluator.py | 57 output_slot: Output slot index as an `int`.
95 output_slot = int(split_items[1])
100 output_slot = int(split_items[1])
103 return device_name, node_name, output_slot, debug_op, exec_index
135 device_name, node_name, output_slot, debug_op, exec_index = (
140 node_name, output_slot, debug_op,
145 "unavailable" % (node_name, output_slot))
|
| /external/tensorflow/tensorflow/core/graph/ |
| costmodel.h | 91 // Records that "output_slot" of "node" has produced tensors of
93 void RecordSize(const Node* node, int output_slot, Bytes bytes);
96 Bytes TotalBytes(const Node* node, int output_slot) const;
99 // output_slot produced by one execution of "node".
100 Bytes SizeEstimate(const Node* node, int output_slot) const;
115 // the tensor generated by "output_slot" of "node". If
116 void RecordMaxMemorySize(const Node* node, int output_slot, Bytes bytes,
120 // Returns the maximum size in bytes of the tensor generated by "output_slot"
122 Bytes MaxMemorySize(const Node* node, int output_slot) const;
125 // generated by "output_slot" of "node"
[all...] |
| costmodel.cc | 249 void CostModel::RecordMaxMemorySize(const Node* node, int output_slot,
255 if (output_slot >= node->num_outputs()) {
257 << ". Got " << output_slot << " but its num_outputs is "
262 auto& current_max = max_mem_usage_[id].output_port_mem[output_slot];
270 max_mem_usage_[id].output_port_shape[output_slot] = tensor_shape;
271 max_mem_usage_[id].output_port_type[output_slot] = dtype;
349 void CostModel::RecordAllocationId(const Node* node, int output_slot,
354 output_port_alloc_ids_[id][output_slot] = alloc_id;
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| debug_utils.py | 28 output_slot=0,
46 output_slot: (`int`) output slot index of the tensor from the watched node.
68 watch.output_slot = output_slot
167 output_slot=slot,
256 output_slot=slot,
|
| grpc_debug_server.py | 38 ["node_name", "output_slot", "debug_op"])
41 def _state_change(new_state, node_name, output_slot, debug_op):
45 state_change.output_slot = output_slot
201 debug_node_key = (state_change.node_name, state_change.output_slot,
206 state_change.output_slot, state_change.debug_op)
211 state_change.output_slot, state_change.debug_op)
217 state_change.node_name, state_change.output_slot,
327 node_name, output_slot, _, debug_op = (
330 DebugWatch(node_name, output_slot, debug_op)
[all...] |
| debug_data.py | 170 def _get_tensor_name(node_name, output_slot):
175 output_slot: Output slot index of the tensor, as an integer.
181 return "%s:%d" % (node_name, output_slot)
184 def _get_tensor_watch_key(node_name, output_slot, debug_op):
190 output_slot: Output slot index of the tensor, as an integer.
198 return "%s:%s" % (_get_tensor_name(node_name, output_slot), debug_op)
276 `node_name`, `output_slot`, `debug_op`, and path to the dump file
303 `node_name`_`output_slot`_`debug_op`_`timestamp`
335 self.output_slot,
407 def output_slot(self) member in class:DebugTensorDatum
[all...] |
| debug_utils_test.py | 76 output_slot, debug_ops and debug_urls.
91 self.assertEqual(expected_output_slot, watch.output_slot)
110 self.assertEqual(1, watch_0.output_slot)
112 self.assertEqual(0, watch_1.output_slot)
135 self.assertEqual(0, watch_0.output_slot)
157 self.assertEqual(0, watch_0.output_slot)
179 self.assertEqual(0, watch_0.output_slot)
|
| source_utils.py | 235 node_name, output_slot, debug_op = watch_key.split(":")
236 tensor_name = "%s:%s" % (node_name, output_slot)
239 dump.get_tensors(node_name, int(output_slot), debug_op))
|
| debug_graphs.py | 44 output_slot = int(name[name.rfind(":") + 1:])
46 return node_name, output_slot
68 _, output_slot = parse_node_or_tensor_name(element_name)
69 return output_slot if output_slot is not None else 0
|
| grpc_debug_test_server.py | 160 output_slot = int(items[1])
162 if ((node_name, output_slot, debug_op) in
229 (node_name, output_slot, debug_op) tuples to toggle the
259 node_name, output_slot, debug_op = watch_key
261 self.request_unwatch(node_name, output_slot, debug_op)
263 self.request_watch(node_name, output_slot, debug_op)
383 (node_name, output_slot, debug_op) tuples to toggle the
|
| debug_gradients.py | 410 node_name, output_slot = debug_graphs.parse_node_or_tensor_name(
414 return dump.get_tensors(node_name, output_slot, "DebugIdentity")
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| debugger_state_interface.cc | 34 strings::StrCat(watch.node_name(), ":", watch.output_slot());
|
| executor.h | 101 typedef std::function<Status(const string& node_name, const int output_slot,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| debug_ops.h | 134 int32 output_slot = 0; local
141 context, strings::safe_strto32(name_items[1], &output_slot),
142 errors::InvalidArgument("Invalid string value for output_slot: \"",
149 new DebugNodeKey(device_name, node_name, output_slot, debug_op_name_));
|
