experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Cleanup headers in save_profile.cc

Browse Source 
PiperOrigin-RevId: 310642299
Change-Id: I0c0f5afa63cc68301eee615ddee717c4870fe5e4
This commit is contained in:
Jose Baiocchi 2020-05-08 15:39:07 -07:00 committed by TensorFlower Gardener
parent aa48faeb08
commit 4f9bc2565f
3 changed files with 21 additions and 9 deletions
tensorflow/core/profiler/rpc/client
BUILDsave_profile.ccsave_profile.h

2
tensorflow/core/profiler/rpc/client/BUILD
View File

@ -27,8 +27,8 @@ cc_library(
deps = [
"//tensorflow/core:framework",
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
"//tensorflow/core/profiler:profiler_service_proto_cc",
"//tensorflow/core/profiler/protobuf:trace_events_proto_cc",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/time",
],

25
tensorflow/core/profiler/rpc/client/save_profile.cc
View File

@ -15,20 +15,28 @@ limitations under the License.
#include "tensorflow/core/profiler/rpc/client/save_profile.h"
#include <cstdio>
#include <ctime>
#include <initializer_list>
#include <memory>
#include <ostream>
#include <sstream>
#include <string>
#include <vector>
#include "absl/strings/match.h"
#include "absl/strings/str_cat.h"
#include "absl/strings/string_view.h"
#include "absl/strings/strip.h"
#include "absl/time/clock.h"
#include "absl/time/time.h"
#include "tensorflow/core/lib/io/compression.h"
#include "tensorflow/core/lib/io/zlib_compression_options.h"
#include "tensorflow/core/lib/io/zlib_outputbuffer.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/errors.h"
#include "tensorflow/core/platform/protobuf.h"
#include "tensorflow/core/profiler/protobuf/trace_events.pb.h"
#include "tensorflow/core/platform/file_system.h"
#include "tensorflow/core/platform/logging.h"
#include "tensorflow/core/platform/status.h"
#include "tensorflow/core/profiler/profiler_service.pb.h"
// Windows.h #defines ERROR, but it is also used in
// tensorflow/core/util/event.proto
#undef ERROR
@ -56,9 +64,9 @@ string ProfilerJoinPathImpl(std::initializer_list<absl::string_view> paths) {
path = absl::StripPrefix(path, kPathSep);
if (absl::EndsWith(result, kPathSep)) {
strings::StrAppend(&result, path);
absl::StrAppend(&result, path);
} else {
strings::StrAppend(&result, kPathSep, path);
absl::StrAppend(&result, kPathSep, path);
}
}
@ -75,7 +83,8 @@ string ProfilerJoinPath(const T&... args) {
constexpr char kProtoTraceFileName[] = "trace";
constexpr char kTfStatsHelperSuffix[] = "tf_stats_helper_result";
Status DumpToolDataToLogDirectory(StringPiece run_dir, const string& host,
Status DumpToolDataToLogDirectory(absl::string_view run_dir,
absl::string_view host,
const ProfileToolData& tool,
std::ostream* os) {
// Don't save the intermediate results for combining the per host tool data.

3
tensorflow/core/profiler/rpc/client/save_profile.h
View File

@ -16,7 +16,10 @@ limitations under the License.
#ifndef TENSORFLOW_CORE_PROFILER_RPC_CLIENT_SAVE_PROFILE_H_
#define TENSORFLOW_CORE_PROFILER_RPC_CLIENT_SAVE_PROFILE_H_
#include <ostream>
#include "tensorflow/core/platform/status.h"
#include "tensorflow/core/platform/types.h"
#include "tensorflow/core/profiler/profiler_service.pb.h"
namespace tensorflow {