1 #pragma once 2 3 #include <torch/csrc/jit/backends/backend_detail.h> 4 namespace torch { 5 namespace jit { 6 class backend_preprocess_register { 7 std::string backend_name_; 8 9 public: backend_preprocess_register(const std::string & name,const detail::BackendPreprocessFunction & preprocess)10 backend_preprocess_register( 11 const std::string& name, 12 const detail::BackendPreprocessFunction& preprocess) 13 : backend_name_(name) { 14 detail::registerBackendPreprocessFunction(name, preprocess); 15 } 16 }; 17 } // namespace jit 18 } // namespace torch 19