aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/platform/regexp.h
blob: ef46a7aca5dde6c1e036f59cab964333acbd6783 (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
30
31
32
33
#ifndef TENSORFLOW_PLATFORM_REGEXP_H_
#define TENSORFLOW_PLATFORM_REGEXP_H_

#include "tensorflow/core/platform/port.h"

#if defined(PLATFORM_GOOGLE) || defined(PLATFORM_GOOGLE_ANDROID)
#include "third_party/re2/re2.h"
namespace tensorflow {
typedef ::StringPiece RegexpStringPiece;
}  // namespace tensorflow

#else

#include "external/re2/re2/re2.h"
namespace tensorflow {
typedef re2::StringPiece RegexpStringPiece;
}  // namespace tensorflow

#endif

namespace tensorflow {

// Conversion to/from the appropriate StringPiece type for using in RE2
inline RegexpStringPiece ToRegexpStringPiece(tensorflow::StringPiece sp) {
  return RegexpStringPiece(sp.data(), sp.size());
}
inline tensorflow::StringPiece FromRegexpStringPiece(RegexpStringPiece sp) {
  return tensorflow::StringPiece(sp.data(), sp.size());
}

}  // namespace tensorflow

#endif  // TENSORFLOW_PLATFORM_REGEXP_H_