Home
last modified time | relevance | path

Searched refs:aliasDb (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_alias_analysis.cpp26 aliasDb = std::make_unique<AliasDb>(graph); in TopologicalMoveTest()
87 return aliasDb->moveBeforeTopologicallyValid(toInsert, insertPoint); in moveBeforeTopologicallyValid()
97 return aliasDb->moveAfterTopologicallyValid(toInsert, insertPoint); in moveAfterTopologicallyValid()
159 std::unique_ptr<AliasDb> aliasDb; member in torch::jit::TopologicalMoveTest
333 AliasDb aliasDb(graph); in TEST() local
335 EXPECT_FALSE(aliasDb.moveAfterTopologicallyValid(c->node(), aMut->node())); in TEST()
336 EXPECT_TRUE(aliasDb.moveAfterTopologicallyValid(d->node(), c->node())); in TEST()
340 aliasDb.moveAfterTopologicallyValid(addsB->node(), aMut->node())); in TEST()
341 EXPECT_TRUE(aliasDb.moveAfterTopologicallyValid(addsB->node(), c->node())); in TEST()
359 AliasDb aliasDb(graph); in TEST() local
[all …]
H A Dtest_custom_operators.cpp171 AliasDb aliasDb(graph); in TEST() local
174 ASSERT_TRUE(aliasDb.writesToAlias(opNode, {input})); in TEST()
176 ASSERT_TRUE(aliasDb.mayAlias(opNode->output(), input)); in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dfusion.cpp22 void createFusionGroups(Block* block, AliasDb* aliasDb, size_t min_size);
34 auto aliasDb = std::make_unique<AliasDb>(graph); in fuseStaticSubgraphs() local
35 createFusionGroups(graph->block(), aliasDb.get(), min_size); in fuseStaticSubgraphs()
116 static bool canMerge(Node* consumer, Node* producer, AliasDb* aliasDb) { in canMerge() argument
126 REQ(aliasDb->couldMoveBeforeTopologically(producer, consumer)); in canMerge()
139 static Node* getOrCreateStaticSubgraph(Node* n, AliasDb* aliasDb) { in getOrCreateStaticSubgraph() argument
145 n, prim::StaticSubgraph, *aliasDb); in getOrCreateStaticSubgraph()
172 AliasDb* aliasDb) { in tryMerge() argument
173 if (!canMerge(fusion_group, to_merge, aliasDb)) { in tryMerge()
189 if (!aliasDb->moveBeforeTopologicallyValid(n, move_point)) { in tryMerge()
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dconstant_pooling.cpp18 const AliasDb& aliasDb) { in ConstantPooling() argument
26 ConstantPooling(block, constants, aliasDb); in ConstantPooling()
49 !aliasDb.safeToChangeAliasingRelationship( in ConstantPooling()
69 AliasDb aliasDb(graph); in ConstantPooling() local
71 ConstantPooling(graph->block(), constants, aliasDb); in ConstantPooling()
H A Dremove_mutation.cpp14 bool MutationRemover::hasSideEffectOrAlias(Value* v, AliasDb* aliasDb) { in hasSideEffectOrAlias() argument
24 aliasDb->mayContainAlias(v->node()->inputs(), v); in hasSideEffectOrAlias()
H A Dcreate_autodiff_subgraphs.cpp34 AliasDb& aliasDb, in SubgraphSlicer() argument
39 aliasDb_(aliasDb), in SubgraphSlicer()
H A Dgraph_fuser.cpp145 GraphFuser(AliasDb* aliasDb, Block* block, bool strict_fuser_check) in GraphFuser()
147 aliasDb_(aliasDb), in GraphFuser()
152 AliasDb* aliasDb, in GraphFuser()
158 aliasDb_(aliasDb), in GraphFuser()
H A Dfrozen_ops_to_mkldnn.cpp113 auto aliasDb = std::make_unique<AliasDb>(graph); in InplaceMKLDNNSubgraph() local
138 if (aliasDb->mayAlias(input, output)) { in InplaceMKLDNNSubgraph()
887 AliasDb& aliasDb) in MKLDNNSubgraphSlicer() argument
888 : block_(block), graph_(std::move(graph)), aliasDb_(aliasDb) {} in MKLDNNSubgraphSlicer()
H A Dremove_mutation.h36 static bool hasSideEffectOrAlias(Value* v, AliasDb* aliasDb);
H A Dfreeze_module.cpp344 std::unique_ptr<AliasDb> aliasDb = in recordMutableAttrs() local
386 if (n->kind() == prim::SetAttr || aliasDb->hasOutputWriters(n)) { in recordMutableAttrs()
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A Ddefer_size_check.cpp19 bool analyzeNode(Node* node, AliasDb& aliasDb) { in analyzeNode() argument
56 if (aliasDb.moveAfterTopologicallyValid(node, shapePreserveOp)) { in analyzeNode()
69 AliasDb aliasDb(graph_); in run() local
71 changed |= analyzeNode(node, aliasDb); in run()
H A Dgraph_fuser.h24 GraphRewriter(Block* block, std::shared_ptr<Graph> graph, AliasDb& aliasDb) in GraphRewriter() argument
27 aliasDb_(aliasDb), in GraphRewriter()
H A Dgraph_helper.cpp507 Node* LlgaGraphHelper::createSingletonSubgraph(Node* n, AliasDb& aliasDb) { in createSingletonSubgraph() argument
512 n, prim::oneDNNFusionGroup, aliasDb); in createSingletonSubgraph()
520 AliasDb& aliasDb) { in mergeNodeIntoSubgraph() argument
542 toMerge, subgraphNode, aliasDb); in mergeNodeIntoSubgraph()
/aosp_15_r20/external/pytorch/torch/csrc/jit/ir/
H A Dalias_analysis.cpp1576 explicit WorkingSet(Node* mover, const AliasDb& aliasDb) in WorkingSet() argument
1577 : aliasDb_(aliasDb), mover_(mover) { in WorkingSet()