Home
last modified time | relevance | path

Searched defs:FunctionExtractor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dfunction_extraction.cpp24 struct FunctionExtractor { struct
26 FunctionExtractor( in FunctionExtractor() function
36 struct ScopeContext {
49 using ScopeCtxPtr = ScopeContext*;
50 using scope_ctx_map = std::unordered_map<ScopePtr, ScopeCtxPtr>;
52 struct FunctionContext {
73 using FunctionCtxPtr = FunctionContext*;
74 using func_ctx_map = std::unordered_map<ScopePtr, FunctionCtxPtr>;
114 std::shared_ptr<Graph> graph_;
115 std::unordered_set<std::string> module_names_;
[all …]