Home
last modified time | relevance | path

Searched refs:output_slot (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/python/debug/cli/
Devaluator_test.py31 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 …]
Devaluator.py95 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))
Danalyzer_cli.py570 dumped_tensor_name = "%s:%d" % (dump.node_name, dump.output_slot)
672 key=lambda x: "%s:%d" % (x.node_name, x.output_slot))
937 node_name, output_slot = debug_graphs.parse_node_or_tensor_name(tensor_name)
950 if output_slot is None:
958 output_slot = list(output_slots)[0]
983 if datum.output_slot == output_slot:
1567 datum.output_slot, datum.debug_op,
1570 command = "pt %s:%d -n %d" % (node_name, datum.output_slot, dump_count)
1655 dumped_tensor_names.append("%s:%d" % (datum.node_name, datum.output_slot))
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_events_monitors.py90 output_slot, argument
99 self._output_slot = output_slot
116 def output_slot(self): member in InfNanAlert
156 output_slot, argument
187 output_slot,
200 output_slot, argument
231 output_slot,
240 output_slot,
254 output_slot,
271 for output_slot, tensor_value in enumerate(tensor_values):
[all …]
Dgrpc_debug_server.py41 def _state_change(new_state, node_name, output_slot, debug_op): argument
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))
394 def request_watch(self, node_name, output_slot, debug_op, breakpoint=False): argument
421 node_name, output_slot, debug_op))
[all …]
Ddebug_events_monitors_test.py150 self.assertEqual(traces[0].output_slot, 0)
152 self.assertEqual(traces[1].output_slot, 0)
155 self.assertEqual(traces[2].output_slot, 0)
163 self.assertEqual(traces[0].output_slot, 0)
165 self.assertEqual(traces[1].output_slot, 0)
168 self.assertEqual(traces[2].output_slot, 0)
179 self.assertEqual(traces[0].output_slot, 0)
185 self.assertEqual(traces[1].output_slot, 0)
191 self.assertEqual(traces[2].output_slot, 1)
197 self.assertEqual(traces[3].output_slot, 0)
[all …]
Ddebug_utils.py28 output_slot=0, argument
68 watch.output_slot = output_slot
176 output_slot=slot,
193 output_slot=-1,
285 output_slot=slot,
Ddebug_data.py171 def _get_tensor_name(node_name, output_slot): argument
182 return "%s:%d" % (node_name, output_slot)
185 def _get_tensor_watch_key(node_name, output_slot, debug_op): argument
199 return "%s:%s" % (_get_tensor_name(node_name, output_slot), debug_op)
336 self.output_slot,
408 def output_slot(self): member in DebugTensorDatum
425 return _get_tensor_name(self.node_name, self.output_slot)
435 return _get_tensor_watch_key(self.node_name, self.output_slot,
567 datum.output_slot].add(datum.debug_op)
865 slot = datum.output_slot
[all …]
Ddebug_utils_test.py95 self.assertEqual(-1, watch.output_slot)
99 self.assertEqual(expected_output_slot, watch.output_slot)
118 self.assertEqual(1, watch_0.output_slot)
120 self.assertEqual(0, watch_1.output_slot)
143 self.assertEqual(0, watch_0.output_slot)
165 self.assertEqual(0, watch_0.output_slot)
187 self.assertEqual(0, watch_0.output_slot)
Ddebug_graphs.py42 output_slot = int(name[name.rfind(":") + 1:])
44 return node_name, output_slot
66 _, output_slot = parse_node_or_tensor_name(element_name)
67 return output_slot if output_slot is not None else 0
Ddebug_events_reader.py573 def get_tensor_id(self, op_name, output_slot): argument
575 return self._op_by_name[op_name].output_tensor_ids[output_slot]
746 def __init__(self, wall_time, locator, op_type, op_name, output_slot, argument
751 self._output_slot = output_slot
763 def output_slot(self): member in GraphExecutionTraceDigest
775 "output_slot": self.output_slot,
807 graph_execution_trace_digest.output_slot,
1059 trace_proto.output_slot,
1339 def symbolic_tensor_id(self, graph_id, op_name, output_slot): argument
1350 return self._graph_by_id[graph_id].get_tensor_id(op_name, output_slot)
[all …]
Dgrpc_debug_test_server.py160 output_slot = int(items[1])
162 if ((node_name, output_slot, debug_op) in
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)
Dsource_utils.py299 node_name, output_slot, debug_op = watch_key.split(":")
300 tensor_name = "%s:%s" % (node_name, output_slot)
303 dump.get_tensors(node_name, int(output_slot), debug_op))
/external/tensorflow/tensorflow/core/debug/
Ddebug_node_key.cc28 const int32 output_slot, const string& debug_op) in DebugNodeKey() argument
31 output_slot(output_slot), in DebugNodeKey()
34 strings::StrCat(node_name, ":", output_slot, ":", debug_op)), in DebugNodeKey()
39 output_slot == other.output_slot && debug_op == other.debug_op); in operator ==()
Ddebug_node_key.h31 const int32 output_slot, const string& debug_op);
43 const int32 output_slot; member
Ddebug_grpc_testlib.cc75 output_slots.push_back(metadata.output_slot()); in SendEvents()
83 metadata.output_slot(), debug_op); in SendEvents()
103 change->set_output_slot(debug_node_key.output_slot); in SendEvents()
Ddebug_io_utils_test.cc65 EXPECT_EQ(0, debug_node_key.output_slot); in TEST_F()
178 ASSERT_EQ(kDebugNodeKey.output_slot, metadata.output_slot()); in TEST_F()
291 ASSERT_EQ(kDebugNodeKey.output_slot, metadata.output_slot()); in TEST_F()
436 ASSERT_EQ(kDebugNodeKey.output_slot, metadata.output_slot()); in TEST_F()
Ddebug_graph_utils.cc81 if (watch.output_slot() == -1) { in InsertNodes()
92 "node name (\"*\"), but got ", watch.output_slot())); in InsertNodes()
96 if (watch.output_slot() < 0) { in InsertNodes()
106 strings::StrCat(watch.node_name(), ":", watch.output_slot()); in InsertNodes()
281 const int output_slot) { in GetCopyNodeName() argument
284 return strings::StrCat("__copy_", node_name, "_", output_slot); in GetCopyNodeName()
Ddebugger_event_metadata.proto8 int32 output_slot = 2; field
/external/tensorflow/tensorflow/core/graph/
Dcostmodel.h101 void RecordSize(const Node* node, int output_slot, Bytes bytes);
104 Bytes TotalBytes(const Node* node, int output_slot) const;
108 Bytes SizeEstimate(const Node* node, int output_slot) const;
124 void RecordMaxMemorySize(const Node* node, int output_slot, Bytes bytes,
130 Bytes MaxMemorySize(const Node* node, int output_slot) const;
135 int output_slot) const;
139 DataType MaxMemoryType(const Node* node, int output_slot) const;
159 void RecordAllocationId(const Node* node, int output_slot, int64 alloc_id);
162 int64 AllocationId(const Node* node, int output_slot) const;
Dcostmodel.cc251 void CostModel::RecordMaxMemorySize(const Node* node, int output_slot, in RecordMaxMemorySize() argument
257 if (output_slot >= node->num_outputs()) { in RecordMaxMemorySize()
259 << ". Got " << output_slot << " but its num_outputs is " in RecordMaxMemorySize()
264 auto& current_max = max_mem_usage_[id].output_port_mem[output_slot]; in RecordMaxMemorySize()
272 max_mem_usage_[id].output_port_shape[output_slot] = tensor_shape; in RecordMaxMemorySize()
273 max_mem_usage_[id].output_port_type[output_slot] = dtype; in RecordMaxMemorySize()
351 void CostModel::RecordAllocationId(const Node* node, int output_slot, in RecordAllocationId() argument
356 output_port_alloc_ids_[id][output_slot] = alloc_id; in RecordAllocationId()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DDebugIdentityV2.pbtxt30 name: "output_slot"
82 name: "output_slot"
141 name: "output_slot"
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_view.cc164 dst_edge->output_slot = e->src_output(); in InitializeNode()
166 const int output_slot = dst_edge->output_slot; in InitializeNode() local
167 if (output_slot >= 0) { in InitializeNode()
168 last_indices[output_slot] = dst_edge; in InitializeNode()
/external/tensorflow/tensorflow/core/protobuf/
Ddebug.proto19 // The semantics of output_slot == -1 is that all outputs of the node
21 // Other negative values of output_slot are invalid and will lead to
23 int32 output_slot = 2; field
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DDebugIdentityV2.pbtxt30 name: "output_slot"

123