Lines Matching refs:debug_node_key
58 Event PrepareChunkEventProto(const DebugNodeKey& debug_node_key,
71 value->set_node_name(debug_node_key.debug_node_name);
75 value->set_tag(debug_node_key.node_name);
79 metadata.set_device(debug_node_key.device_name);
80 metadata.set_output_slot(debug_node_key.output_slot);
98 << "The debug_node_name is " << debug_node_key.debug_node_name
123 Status WrapStringTensorAsEvents(const DebugNodeKey& debug_node_key,
143 debug_node_key.debug_node_name,
156 Event event = PrepareChunkEventProto(debug_node_key, wall_time_us,
183 Status WrapTensorAsEvents(const DebugNodeKey& debug_node_key,
195 debug_node_key, wall_time_us, chunk_size_limit, &tensor_proto, events));
210 Event event = PrepareChunkEventProto(debug_node_key, wall_time_us,
410 Status DebugIO::PublishDebugTensor(const DebugNodeKey& debug_node_key,
422 debug_node_key, tensor, wall_time_us, dump_root_dir, nullptr);
430 debug_node_key, tensor, wall_time_us, url, gated_grpc);
444 (*callback)(debug_node_key, tensor);
466 Status DebugIO::PublishDebugTensor(const DebugNodeKey& debug_node_key,
470 return PublishDebugTensor(debug_node_key, tensor, wall_time_us, debug_urls,
577 Status DebugFileIO::DumpTensorToDir(const DebugNodeKey& debug_node_key,
583 GetDumpFilePath(dump_root_dir, debug_node_key, wall_time_us);
589 return DumpTensorToEventFile(debug_node_key, tensor, wall_time_us, file_path);
593 const DebugNodeKey& debug_node_key,
596 io::JoinPath(dump_root_dir, debug_node_key.device_path,
597 strings::StrCat(debug_node_key.node_name, "_",
598 debug_node_key.output_slot, "_",
599 debug_node_key.debug_op)),
628 Status DebugFileIO::DumpTensorToEventFile(const DebugNodeKey& debug_node_key,
634 WrapTensorAsEvents(debug_node_key, tensor, wall_time_us, 0, &events));
755 const DebugNodeKey& debug_node_key, const Tensor& tensor,
759 !IsReadGateOpen(grpc_stream_url, debug_node_key.debug_node_name)) {
763 TF_RETURN_IF_ERROR(WrapTensorAsEvents(debug_node_key, tensor, wall_time_us,
769 if (IsWriteGateOpen(grpc_stream_url, debug_node_key.debug_node_name)) {
