Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/
H A Dinterface.cpp26 int64_t registerFusion(const Node* fusion_group) { in registerFusion()
59 Node* fusion_group = wrapper_graph->insertNode( in debugLaunchGraph() local
81 Node* fusion_group = wrapper_graph->insertNode( in debugGetFusedKernelCode() local
H A Dcompiler.cpp178 int64_t registerFusion(const Node* fusion_group) { in registerFusion()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dtensorexpr_fuser.cpp429 std::unordered_map<Value*, Value*> buildShapeExpressions(Node* fusion_group) { in buildShapeExpressions()
522 void removeOutputsUsedOnlyInSize(Node* fusion_group) { in removeOutputsUsedOnlyInSize()
698 Node* fusion_group = initial_fusion_groups[i]; in createFusionGroups() local
777 std::optional<Node*> tryMerge(Node* fusion_group, Node* to_merge) { in tryMerge()
H A Dgraph_fuser.cpp880 auto fusion_group = tryFuse(consumer, producer); in scanNode() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dfusion.cpp170 Node* fusion_group, in tryMerge()
307 Node* fusion_group = initial_fusion_groups[i]; in createFusionGroups() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/
H A Dsubgraph_utils.cpp110 Node* fusion_group = merge_fn(); in executeSubgraphMergeAndUpdateAliasing() local
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_misc.cpp384 auto fusion_group = in TEST() local