| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_node_key.cc | 23 const char* const DebugNodeKey::kMetadataFilePrefix = "_tfdbg_";
25 const char* const DebugNodeKey::kDeviceTag = "device_";
27 DebugNodeKey::DebugNodeKey(const string& device_name, const string& node_name, in DebugNodeKey() function in tensorflow::DebugNodeKey
37 bool DebugNodeKey::operator==(const DebugNodeKey& other) const { in operator ==()
42 bool DebugNodeKey::operator!=(const DebugNodeKey& other) const { in operator !=()
46 const string DebugNodeKey::DeviceNameToDevicePath(const string& device_name) { in DeviceNameToDevicePath()
|
| D | debug_node_key.h | 26 struct DebugNodeKey { struct
30 DebugNodeKey(const string& device_name, const string& node_name, argument
38 bool operator==(const DebugNodeKey& other) const;
39 bool operator!=(const DebugNodeKey& other) const;
|
| D | debug_io_utils_test.cc | 60 DebugNodeKey debug_node_key("/job:worker/replica:1/task:0/device:GPU:2", in TEST_F()
73 const DebugNodeKey debug_node_key_1("/job:worker/replica:1/task:0/gpu:2", in TEST_F()
75 const DebugNodeKey debug_node_key_2("/job:worker/replica:1/task:0/gpu:2", in TEST_F()
77 const DebugNodeKey debug_node_key_3("/job:worker/replica:1/task:0/gpu:2", in TEST_F()
79 const DebugNodeKey debug_node_key_4("/job:worker/replica:1/task:0/gpu:2", in TEST_F()
89 const DebugNodeKey debug_node_key_1("/job:worker/replica:1/task:0/gpu:2", in TEST_F()
91 const DebugNodeKey debug_node_key_2("/job:worker/replica:1/task:0/gpu:2", in TEST_F()
93 const DebugNodeKey debug_node_key_3("/job:worker/replica:1/task:0/gpu:2", in TEST_F()
96 std::unordered_set<DebugNodeKey> keys; in TEST_F()
113 const DebugNodeKey kDebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", in TEST_F()
[all …]
|
| D | debug_io_utils.h | 78 static Status PublishDebugTensor(const DebugNodeKey& debug_node_key,
85 static Status PublishDebugTensor(const DebugNodeKey& debug_node_key,
168 static Status DumpTensorToDir(const DebugNodeKey& debug_node_key,
183 const DebugNodeKey& debug_node_key,
218 static Status DumpTensorToEventFile(const DebugNodeKey& debug_node_key,
243 struct hash<::tensorflow::DebugNodeKey> {
244 size_t operator()(const ::tensorflow::DebugNodeKey& k) const {
341 static Status SendTensorThroughGrpcStream(const DebugNodeKey& debug_node_key,
|
| D | debug_grpc_testlib.h | 49 const DebugNodeKey& debug_node_key);
63 std::vector<DebugNodeKey> debug_node_keys_ TF_GUARDED_BY(states_mu_);
67 std::unordered_set<DebugNodeKey> write_enabled_debug_node_keys_;
|
| D | debug_grpc_testlib.cc | 82 const DebugNodeKey debug_node_key(metadata.device(), node_name, in SendEvents()
101 const DebugNodeKey& debug_node_key = debug_node_keys_[i]; in SendEvents()
133 const DebugNodeKey& debug_node_key) { in RequestDebugOpStateChangeAtNextStream()
174 DebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", "prep_node", 0, in PollTillFirstRequestSucceeds()
|
| D | debug_grpc_io_utils_test.cc | 91 DebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", "foo_tensor", 0, in TEST_F()
112 const DebugNodeKey kDebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", in TEST_F()
132 const DebugNodeKey kDebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", in TEST_F()
151 const DebugNodeKey kDebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", in TEST_F()
167 const DebugNodeKey kDebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", in TEST_F()
213 DebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", in TEST_F()
262 const DebugNodeKey kDebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", in TEST_F()
310 const DebugNodeKey kDebugNodeKey("/job:localhost/replica:0/task:0/cpu:0", in TEST_F()
|
| D | debug_io_utils.cc | 61 Event PrepareChunkEventProto(const DebugNodeKey& debug_node_key, in PrepareChunkEventProto()
126 Status WrapStringTensorAsEvents(const DebugNodeKey& debug_node_key, in WrapStringTensorAsEvents()
186 Status WrapTensorAsEvents(const DebugNodeKey& debug_node_key, in WrapTensorAsEvents()
400 DebugNodeKey::kMetadataFilePrefix, in PublishDebugMetadata()
414 Status DebugIO::PublishDebugTensor(const DebugNodeKey& debug_node_key, in PublishDebugTensor()
483 Status DebugIO::PublishDebugTensor(const DebugNodeKey& debug_node_key, in PublishDebugTensor()
509 DebugNodeKey::DeviceNameToDevicePath(device_name)); in PublishGraph()
512 strings::StrCat(DebugNodeKey::kMetadataFilePrefix, DebugIO::kGraphTag, in PublishGraph()
594 Status DebugFileIO::DumpTensorToDir(const DebugNodeKey& debug_node_key, in DumpTensorToDir()
610 const DebugNodeKey& debug_node_key, in GetDumpFilePath()
[all …]
|
| D | debug_callback_registry.h | 42 using EventCallback = std::function<void(const DebugNodeKey&, const Tensor&)>;
|
| D | grpc_session_debug_test.cc | 193 LoadTensorDumps(io::JoinPath(DebugNodeKey::DeviceNameToDevicePath( in TEST_F()
275 io::JoinPath(DebugNodeKey::DeviceNameToDevicePath(a_dev.name()), in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | debug_ops_test.cc | 107 DebugNodeKey::kMetadataFilePrefix, DebugNodeKey::kDeviceTag); in TEST_F()
|
| D | debug_ops.h | 148 new DebugNodeKey(device_name, node_name, output_slot, debug_op_name_)); in BaseDebugOp()
199 std::unique_ptr<DebugNodeKey> debug_watch_key_;
|