Home
last modified time | relevance | path

Searched refs:BackendPreprocessFunction (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/
H A Dbackend_detail.h21 using BackendPreprocessFunction = std::function<c10::IValue( variable
28 const BackendPreprocessFunction& preprocess);
32 BackendPreprocessFunction getBackendPreprocessFunction(const std::string& name);
H A Dbackend_detail.cpp64 std::unordered_map<std::string, BackendPreprocessFunction>&
66 static std::unordered_map<std::string, BackendPreprocessFunction> in backendPreprocessFunctions()
78 const BackendPreprocessFunction& preprocess) { in registerBackendPreprocessFunction()
87 BackendPreprocessFunction getBackendPreprocessFunction( in getBackendPreprocessFunction()
H A Dbackend_preprocess.h12 const detail::BackendPreprocessFunction& preprocess) in backend_preprocess_register()