/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/backends
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/backends/backend_preprocess.h (438B)
#pragma once
#include
namespace torch {
namespace jit {
class backend_preprocess_register {
std::string backend_name_;
public:
backend_preprocess_register(
const std::string& name,
const detail::BackendPreprocessFunction& preprocess)
: backend_name_(name) {
detail::registerBackendPreprocessFunction(name, preprocess);
}
};
} // namespace jit
} // namespace torch