xref: /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/external_functions_registry.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/csrc/jit/tensorexpr/external_functions_registry.h>
2 
3 namespace torch::jit::tensorexpr {
4 
getNNCFunctionRegistry()5 std::unordered_map<std::string, NNCExternalFunction>& getNNCFunctionRegistry() {
6   static std::unordered_map<std::string, NNCExternalFunction> func_registry_;
7   return func_registry_;
8 }
9 
10 } // namespace torch::jit::tensorexpr
11