aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/stream_executor/lib/env.h
blob: 74b50ad42d44dc403a59fecff8211cb62233e27b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#ifndef TENSORFLOW_STREAM_EXECUTOR_LIB_ENV_H_
#define TENSORFLOW_STREAM_EXECUTOR_LIB_ENV_H_

#include "tensorflow/core/public/env.h"
#include "tensorflow/stream_executor/lib/stringpiece.h"
#include "tensorflow/stream_executor/platform/port.h"

namespace perftools {
namespace gputools {
namespace port {

using tensorflow::Env;
using tensorflow::ReadFileToString;
using tensorflow::Thread;
using tensorflow::WriteStringToFile;

inline bool FileExists(const string& filename) {
  return Env::Default()->FileExists(filename);
}

inline bool FileExists(const port::StringPiece& filename) {
  return Env::Default()->FileExists(filename.ToString());
}

}  // namespace port
}  // namespace gputools
}  // namespace perftools

#endif  // TENSORFLOW_STREAM_EXECUTOR_LIB_ENV_H_