Searched defs:aliasDb (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_alias_analysis.cpp | 159 std::unique_ptr<AliasDb> aliasDb; member in torch::jit::TopologicalMoveTest
333 AliasDb aliasDb(graph); in TEST() local
359 AliasDb aliasDb(graph); in TEST() local
373 AliasDb aliasDb(graph); in TEST() local
430 AliasDb aliasDb(graph); in TEST() local
448 AliasDb aliasDb(graph); in TEST() local
471 AliasDb aliasDb(graph); in TEST() local
528 AliasDb aliasDb(graph); in TEST_P() local
564 AliasDb aliasDb(graph); in TEST_P() local
604 AliasDb aliasDb(graph); in TEST_P() local
[all …]
|
| H A D | test_custom_operators.cpp | 171 AliasDb aliasDb(graph); in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | fusion.cpp | 34 auto aliasDb = std::make_unique<AliasDb>(graph); in fuseStaticSubgraphs() local
116 static bool canMerge(Node* consumer, Node* producer, AliasDb* aliasDb) { in canMerge()
139 static Node* getOrCreateStaticSubgraph(Node* n, AliasDb* aliasDb) { in getOrCreateStaticSubgraph()
172 AliasDb* aliasDb) { in tryMerge()
210 AliasDb* aliasDb) { in createFusionGroup()
233 AliasDb* aliasDb) { in scanNode()
269 void createFusionGroups(Block* block, AliasDb* aliasDb, size_t min_size) { in createFusionGroups()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | constant_pooling.cpp | 18 const AliasDb& aliasDb) { in ConstantPooling()
69 AliasDb aliasDb(graph); in ConstantPooling() local
|
| H A D | frozen_ops_to_mkldnn.cpp | 113 auto aliasDb = std::make_unique<AliasDb>(graph); in InplaceMKLDNNSubgraph() local
887 AliasDb& aliasDb) in MKLDNNSubgraphSlicer()
|
| H A D | remove_mutation.cpp | 14 bool MutationRemover::hasSideEffectOrAlias(Value* v, AliasDb* aliasDb) { in hasSideEffectOrAlias()
|
| H A D | create_autodiff_subgraphs.cpp | 34 AliasDb& aliasDb, in SubgraphSlicer()
|
| H A D | freeze_module.cpp | 344 std::unique_ptr<AliasDb> aliasDb = in recordMutableAttrs() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
| H A D | defer_size_check.cpp | 19 bool analyzeNode(Node* node, AliasDb& aliasDb) { in analyzeNode()
69 AliasDb aliasDb(graph_); in run() local
|
| H A D | graph_helper.cpp | 507 Node* LlgaGraphHelper::createSingletonSubgraph(Node* n, AliasDb& aliasDb) { in createSingletonSubgraph()
520 AliasDb& aliasDb) { in mergeNodeIntoSubgraph()
|
| H A D | graph_fuser.h | 24 GraphRewriter(Block* block, std::shared_ptr<Graph> graph, AliasDb& aliasDb) in GraphRewriter()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | alias_analysis.cpp | 1576 explicit WorkingSet(Node* mover, const AliasDb& aliasDb) in WorkingSet()
|