/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_node_key.h | 31 const int32 output_slot, const string& debug_op); 43 const int32 output_slot; member in struct:tensorflow::DebugNodeKey
|
debug_service.proto | 39 int32 output_slot = 3;
|
debug_grpc_testlib.cc | 75 output_slots.push_back(metadata.output_slot()); 83 metadata.output_slot(), debug_op); 103 change->set_output_slot(debug_node_key.output_slot);
|
debug_graph_utils.h | 86 const int output_slot);
|
debug_io_utils_test.cc | 65 EXPECT_EQ(0, debug_node_key.output_slot); 178 ASSERT_EQ(kDebugNodeKey.output_slot, metadata.output_slot()); 291 ASSERT_EQ(kDebugNodeKey.output_slot, metadata.output_slot()); 436 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);
|
debug_grpc_io_utils_test.cc | 124 EXPECT_EQ(kDebugNodeKey.output_slot, server_data.server->output_slots[0]); 144 EXPECT_EQ(kDebugNodeKey.output_slot, server_data_.server->output_slots[0]); 299 EXPECT_EQ(kDebugNodeKey.output_slot, 349 EXPECT_EQ(kDebugNodeKey.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))
|
analyzer_cli.py | 390 1) in the simplest case, use <node_name>:<output_slot>, e.g., 394 <node_name>:<output_slot>:<debug_op>, e.g., 398 <device_name>:<node_name>:<output_slot>[:<debug_op>], e.g., 572 dumped_tensor_name = "%s:%d" % (dump.node_name, dump.output_slot) 674 key=lambda x: "%s:%d" % (x.node_name, x.output_slot)) [all...] |
/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 170 output_slot=slot, 263 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 | 171 def _get_tensor_name(node_name, output_slot): 176 output_slot: Output slot index of the tensor, as an integer. 182 return "%s:%d" % (node_name, output_slot) 185 def _get_tensor_watch_key(node_name, output_slot, debug_op): 191 output_slot: Output slot index of the tensor, as an integer. 199 return "%s:%s" % (_get_tensor_name(node_name, output_slot), debug_op) 277 `node_name`, `output_slot`, `debug_op`, and path to the dump file 304 `node_name`_`output_slot`_`debug_op`_`timestamp` 336 self.output_slot, 408 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) 384 (node_name, output_slot, debug_op) tuples to toggle the
|
debug_gradients.py | 411 node_name, output_slot = debug_graphs.parse_node_or_tensor_name( 415 return dump.get_tensors(node_name, output_slot, "DebugIdentity")
|
debug_data_test.py | 124 self.assertEqual(2, datum.output_slot) 133 datum.output_slot, 139 datum.output_slot,
|
/external/tensorflow/tensorflow/core/common_runtime/ |
debugger_state_interface.cc | 34 strings::StrCat(watch.node_name(), ":", watch.output_slot());
|
step_stats_collector.cc | 356 int output_slot = output.slot(); local 357 cm->RecordMaxMemorySize(node, output_slot, 362 node->output_types()[output_slot]); 363 cm->RecordAllocationId(node, 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_));
|