diff options
| author |  Yu-Cheng Ling <ycling@google.com> | 2018-01-17 12:28:46 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-17 12:32:50 -0800 |
| commit | 36635f4a389c812bb328821e5e533feeef7d26ed (patch) | |
| tree | 4ce54e3a191ffc175296663a6529819c367cc90b /tensorflow/contrib/lite/testing/split.h | |
| parent | aabe9698daf5004ec90f04e2da5b71b51b010b6d (diff) | |
Make int64 inputs workable, and testable in generate_examples_test.
PiperOrigin-RevId: 182252829
Diffstat (limited to 'tensorflow/contrib/lite/testing/split.h')
| -rw-r--r-- | tensorflow/contrib/lite/testing/split.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/testing/split.h b/tensorflow/contrib/lite/testing/split.h index 24071442e8..cfc1e929e9 100644 --- a/tensorflow/contrib/lite/testing/split.h +++ b/tensorflow/contrib/lite/testing/split.h @@ -54,6 +54,15 @@ inline std::vector<int> Split(const string& s, const string& delimiter) { } template <> +inline std::vector<int64_t> Split(const string& s, const string& delimiter) { + std::vector<int64_t> fields; + for (const auto& p : SplitToPos(s, delimiter)) { + fields.push_back(strtoll(s.data() + p.first, nullptr, 10)); + } + return fields; +} + +template <> inline std::vector<float> Split(const string& s, const string& delimiter) { std::vector<float> fields; for (const auto& p : SplitToPos(s, delimiter)) { |