Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dbatch_mm.cpp255 static void BatchMMTreeReduce(Block* block, AliasDb& alias_db) { in BatchMMTreeReduce()
374 AliasDb& alias_db) { in gatherIndependentMMUses()
417 static void BatchMMSide(Block* block, AliasDb& alias_db) { in BatchMMSide()
481 AliasDb alias_db(graph); in BatchMM() local
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_module.cc1128 AliasDb alias_db(graph); in TEST() local
1168 AliasDb alias_db(graph); in TEST() local
1215 AliasDb alias_db(graph); in TEST() local
1250 AliasDb alias_db(graph); in TEST() local
1271 AliasDb alias_db(graph); in TEST() local
1358 AliasDb alias_db(graph); in testAssignStorageToManagedTensors() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.cpp399 const AliasDb& alias_db, in ManagedTensorRanges()
593 AliasDb alias_db(graph_, /*isFrozen=*/false); in StaticModule() local
679 const AliasDb& alias_db, in prepareFunctionsAndConstants()
712 const AliasDb& alias_db, in prepareStaticNodeInfos()
803 const AliasDb& alias_db, in prepare_for_memory_planner()
H A Dimpl.h366 void init_value_group(const AliasDb& alias_db) { in init_value_group()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/
H A Dsubgraph_utils.cpp183 AliasDb alias_db(std::move(subgraph)); in buildAliasedSets() local
440 AliasDb alias_db(subgraph); in unmergeOutputsAlisingInputs() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/dbr_quantization/
H A Dremove_redundant_aliases.cpp17 AliasDb alias_db(g, is_frozen, descend_function_calls); in DBRQuantRemoveRedundantAliasesImpl() local