Home
last modified time | relevance | path

Searched refs:output_slot (Results 1 – 25 of 52) 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))
936 node_name, output_slot = debug_graphs.parse_node_or_tensor_name(tensor_name)
949 if output_slot is None:
957 output_slot = list(output_slots)[0]
982 if datum.output_slot == output_slot:
1566 datum.output_slot, datum.debug_op,
1569 command = "pt %s:%d -n %d" % (node_name, datum.output_slot, dump_count)
1654 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
214 output_slot,
223 output_slot,
244 for output_slot, tensor_value in enumerate(tensor_values):
246 tensor_value, execution.wall_time, execution.op_type, output_slot,
[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_utils.py28 output_slot=0, argument
68 watch.output_slot = output_slot
174 output_slot=slot,
191 output_slot=-1,
284 output_slot=slot,
Ddebug_events_monitors_test.py142 self.assertEqual(traces[0].output_slot, 0)
144 self.assertEqual(traces[1].output_slot, 0)
147 self.assertEqual(traces[2].output_slot, 0)
155 self.assertEqual(traces[0].output_slot, 0)
161 self.assertEqual(traces[1].output_slot, 0)
167 self.assertEqual(traces[2].output_slot, 1)
173 self.assertEqual(traces[3].output_slot, 0)
195 self.assertEqual(alert.output_slot, 1)
230 self.assertEqual(alert.output_slot, 1)
260 self.assertEqual(alert.output_slot, 0)
[all …]
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.py94 self.assertEqual(-1, watch.output_slot)
98 self.assertEqual(expected_output_slot, watch.output_slot)
117 self.assertEqual(1, watch_0.output_slot)
119 self.assertEqual(0, watch_1.output_slot)
142 self.assertEqual(0, watch_0.output_slot)
164 self.assertEqual(0, watch_0.output_slot)
186 self.assertEqual(0, watch_0.output_slot)
Ddumping_callback.py334 for output_slot, tensor in enumerate(tensors):
354 output_slot=output_slot,
364 for output_slot, tensor in enumerate(tensors):
381 tensor_id=tensor_ids[output_slot],
386 output_slot=output_slot,
394 for output_slot, tensor in enumerate(tensors):
406 output_slot=output_slot,
Ddebug_events_reader.py428 def get_tensor_id(self, op_name, output_slot): argument
430 return self._op_by_name[op_name].output_tensor_ids[output_slot]
554 output_slot, argument
559 self._output_slot = output_slot
571 def output_slot(self): member in GraphExecutionTraceDigest
583 "output_slot": self.output_slot,
615 graph_execution_trace_digest.output_slot,
869 trace_proto.output_slot,
1138 def symbolic_tensor_id(self, graph_id, op_name, output_slot): argument
1149 return self._graph_by_id[graph_id].get_tensor_id(op_name, output_slot)
[all …]
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
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.py303 node_name, output_slot, debug_op = watch_key.split(":")
304 tensor_name = "%s:%s" % (node_name, output_slot)
307 dump.get_tensors(node_name, int(output_slot), debug_op))
Ddebug_gradients.py411 node_name, output_slot = debug_graphs.parse_node_or_tensor_name(
415 return dump.get_tensors(node_name, output_slot, "DebugIdentity")
/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
Ddebug_grpc_io_utils_test.cc124 EXPECT_EQ(kDebugNodeKey.output_slot, server_data.server->output_slots[0]); in TEST_F()
144 EXPECT_EQ(kDebugNodeKey.output_slot, server_data_.server->output_slots[0]); in TEST_F()
299 EXPECT_EQ(kDebugNodeKey.output_slot, in TEST_F()
349 EXPECT_EQ(kDebugNodeKey.output_slot, in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dcostmodel.h93 void RecordSize(const Node* node, int output_slot, Bytes bytes);
96 Bytes TotalBytes(const Node* node, int output_slot) const;
100 Bytes SizeEstimate(const Node* node, int output_slot) const;
116 void RecordMaxMemorySize(const Node* node, int output_slot, Bytes bytes,
122 Bytes MaxMemorySize(const Node* node, int output_slot) const;
127 int output_slot) const;
131 DataType MaxMemoryType(const Node* node, int output_slot) const;
151 void RecordAllocationId(const Node* node, int output_slot, int64 alloc_id);
154 int64 AllocationId(const Node* node, int output_slot) const;
Dcostmodel.cc249 void CostModel::RecordMaxMemorySize(const Node* node, int output_slot, in RecordMaxMemorySize() argument
255 if (output_slot >= node->num_outputs()) { in RecordMaxMemorySize()
257 << ". Got " << output_slot << " but its num_outputs is " in RecordMaxMemorySize()
262 auto& current_max = max_mem_usage_[id].output_port_mem[output_slot]; in RecordMaxMemorySize()
270 max_mem_usage_[id].output_port_shape[output_slot] = tensor_shape; in RecordMaxMemorySize()
271 max_mem_usage_[id].output_port_type[output_slot] = dtype; in RecordMaxMemorySize()
349 void CostModel::RecordAllocationId(const Node* node, int output_slot, in RecordAllocationId() argument
354 output_port_alloc_ids_[id][output_slot] = alloc_id; in RecordAllocationId()
/external/tensorflow/tensorflow/core/protobuf/
Ddebug.proto18 // The semantics of output_slot == -1 is that all outputs of the node
20 // Other negative values of output_slot are invalid and will lead to
22 int32 output_slot = 2; field

123