mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
1a73255102
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/138976 Approved by: https://github.com/Skylion007
17 lines
413 B
C++
17 lines
413 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/jit/backends/backend_detail.h>
|
|
namespace torch::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 torch::jit
|