mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
71ca600af9
Summary: Since caffe2 and torch have been consolidated, CAFFE2_API should be merged with TORCH_API. Addresses a TODO. Manually edited some references of the removed `CAFFE2_API`: * `CONTRIBUTING.md` * `caffe2/proto/CMakeLists.txt` * `cmake/ProtoBuf.cmake` * `c10/macros/Export.h` * `torch/csrc/WindowsTorchApiMacro.h` Pull Request resolved: https://github.com/pytorch/pytorch/pull/49496 Reviewed By: malfet, samestep Differential Revision: D25600726 Pulled By: janeyx99 fbshipit-source-id: 7e068d959e397ac183c097d7e9a9afeca5ddd782
51 lines
1.2 KiB
C++
51 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <algorithm>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "caffe2/core/common.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
TORCH_API std::vector<std::string>
|
|
split(char separator, const std::string& string, bool ignore_empty = false);
|
|
|
|
TORCH_API std::string trim(const std::string& str);
|
|
|
|
TORCH_API size_t editDistance(
|
|
const std::string& s1,
|
|
const std::string& s2,
|
|
size_t max_distance = 0);
|
|
|
|
TORCH_API inline bool StartsWith(
|
|
const std::string& str,
|
|
const std::string& prefix) {
|
|
return str.length() >= prefix.length() &&
|
|
std::mismatch(prefix.begin(), prefix.end(), str.begin()).first ==
|
|
prefix.end();
|
|
}
|
|
|
|
TORCH_API inline bool EndsWith(
|
|
const std::string& full,
|
|
const std::string& ending) {
|
|
if (full.length() >= ending.length()) {
|
|
return (
|
|
0 ==
|
|
full.compare(full.length() - ending.length(), ending.length(), ending));
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
TORCH_API int32_t editDistanceHelper(
|
|
const char* s1,
|
|
size_t s1_len,
|
|
const char* s2,
|
|
size_t s2_len,
|
|
std::vector<size_t>& current,
|
|
std::vector<size_t>& previous,
|
|
std::vector<size_t>& previous1,
|
|
size_t max_distance);
|
|
} // namespace caffe2
|