Home
last modified time | relevance | path

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

12

/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.py572 dumped_tensor_name = "%s:%d" % (dump.node_name, dump.output_slot)
674 key=lambda x: "%s:%d" % (x.node_name, x.output_slot))
938 node_name, output_slot = debug_graphs.parse_node_or_tensor_name(tensor_name)
951 if output_slot is None:
959 output_slot = list(output_slots)[0]
984 if datum.output_slot == output_slot:
1569 datum.output_slot, datum.debug_op,
1572 command = "pt %s:%d -n %d" % (node_name, datum.output_slot, dump_count)
1657 dumped_tensor_names.append("%s:%d" % (datum.node_name, datum.output_slot))
/external/tensorflow/tensorflow/core/debug/
Ddebug_node_key.cc27 const int32 output_slot, const string& debug_op) in DebugNodeKey() argument
30 output_slot(output_slot), in DebugNodeKey()
33 strings::StrCat(node_name, ":", output_slot, ":", debug_op)), in DebugNodeKey()
38 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_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_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_graph_utils.cc68 if (watch.output_slot() < 0) { in InsertNodes()
79 strings::StrCat(watch.node_name(), ":", watch.output_slot()); in InsertNodes()
248 const int output_slot) { in GetCopyNodeName() argument
251 return strings::StrCat("__copy_", node_name, "_", output_slot); in GetCopyNodeName()
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()
Ddebugger_event_metadata.proto8 int32 output_slot = 2; field
Ddebug_graph_utils.h86 const int output_slot);
/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/python/debug/lib/
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))
391 def request_watch(self, node_name, output_slot, debug_op, breakpoint=False): argument
418 node_name, output_slot, debug_op))
[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.py28 output_slot=0, argument
68 watch.output_slot = output_slot
170 output_slot=slot,
263 output_slot=slot,
Ddebug_utils_test.py91 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)
Ddebug_graphs.py44 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
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.py235 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))
Ddebug_gradients.py411 node_name, output_slot = debug_graphs.parse_node_or_tensor_name(
415 return dump.get_tensors(node_name, output_slot, "DebugIdentity")
Ddebug_data_test.py124 self.assertEqual(2, datum.output_slot)
133 datum.output_slot,
139 datum.output_slot,
/external/tensorflow/tensorflow/core/kernels/
Ddebug_ops.h134 int32 output_slot = 0; in BaseDebugOp() local
141 context, strings::safe_strto32(name_items[1], &output_slot), in BaseDebugOp()
149 new DebugNodeKey(device_name, node_name, output_slot, debug_op_name_)); in BaseDebugOp()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc356 int output_slot = output.slot(); in BuildCostModel() local
357 cm->RecordMaxMemorySize(node, output_slot, in BuildCostModel()
362 node->output_types()[output_slot]); in BuildCostModel()
363 cm->RecordAllocationId(node, output_slot, in BuildCostModel()
/external/tensorflow/tensorflow/core/protobuf/
Ddebug.proto16 // The semantics of output_slot == -1 is that the node is only watched for
20 int32 output_slot = 2; field

12