Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dgraph_fuser.cpp125 struct GraphFuser { struct
126 using FusionCallback = std::function<bool(GraphFuser*, Node*)>;
128 Block* block_;
129 AliasDb* aliasDb_;
130 std::shared_ptr<Graph> graph_;
131 FusionCallback callback_ = [](GraphFuser* gf, Node* n) { in __anon688491630202()
134 Symbol kind_ = prim::FusionGroup;
135 bool strict_fuser_check_ = false;
143 size_t subgraph_arg_limit_ = 128;
145 GraphFuser(AliasDb* aliasDb, Block* block, bool strict_fuser_check) in GraphFuser() function
[all …]