Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A Dguard_shape.cpp21 std::vector<Node*> fusion_groups; in prepareFusionGroupAndGuardOutputs() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dtensorexpr_fuser.cpp1252 std::vector<Node*> fusion_groups; in prepareFusionGroupAndGuardOutputs() local
1271 std::vector<Node*> fusion_groups; in generalizeFusionGroups() local