Home
last modified time | relevance | path

Searched refs:DepthFirstGraphNodeIterator (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/
H A Dxnnpack_graph_builder.cpp108 DepthFirstGraphNodeIterator it(graph); in checkOpsToDelegate()
163 DepthFirstGraphNodeIterator it(graph); in defineAllNodes()
289 DepthFirstGraphNodeIterator it(graph); in gatherTensorValues()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dgraph_iterator.h6 class DepthFirstGraphNodeIterator {
11 explicit DepthFirstGraphNodeIterator(std::shared_ptr<Graph>& graph) in DepthFirstGraphNodeIterator() function
H A Dsymbolic_shape_registry.cpp123 DepthFirstGraphNodeIterator graph_it(graph); in checkForWhileLoop()
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_graph_iterator.cpp43 DepthFirstGraphNodeIterator graph_it(graph); in traverse_depth_first()
79 DepthFirstGraphNodeIterator graph_it(graph); in TEST()
H A Dtest_shape_analysis.cpp28 DepthFirstGraphNodeIterator graph_it(g); in findNode()
135 DepthFirstGraphNodeIterator graph_it(g); in TEST()
H A Dtest_alias_analysis.cpp518 DepthFirstGraphNodeIterator it(graph); in TEST_P()
554 DepthFirstGraphNodeIterator it(graph); in TEST_P()
594 DepthFirstGraphNodeIterator it(graph); in TEST_P()
H A Dtest_autodiff.cpp328 DepthFirstGraphNodeIterator it(optimized_plan.graph); in TEST_F()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dpasses.cpp20 DepthFirstGraphNodeIterator graph_it(graph); in graphHasOp()
613 DepthFirstGraphNodeIterator graph_it(graph); in ReplaceWithMaybeCopy()
689 DepthFirstGraphNodeIterator graph_it(graph); in ReplaceWithCopyImpl()
804 DepthFirstGraphNodeIterator graph_it(graph); in EliminateTrivialEquallySplit()
901 DepthFirstGraphNodeIterator graph_it(graph); in FuseListUnpack()
1348 DepthFirstGraphNodeIterator it(graph); in EliminateNoOpSlice()
H A Dfusion.cpp325 DepthFirstGraphNodeIterator it(graph); in inlineFallbackGraphs()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dfinalize.cpp49 DepthFirstGraphNodeIterator it(g); in removePackedParamInsertionAndFPWeightsSetAttr()
79 DepthFirstGraphNodeIterator it(g); in removeObserverCallMethods()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Drefine_tuple_types.cpp34 DepthFirstGraphNodeIterator it(graph); in RefineTupleTypes()
H A Deliminate_no_ops.cpp59 auto graph_it = DepthFirstGraphNodeIterator(graph); in EliminateNoOps()
H A Dadd_if_then_else.cpp24 DepthFirstGraphNodeIterator graph_it(graph); in AddIfThenElseOp()
H A Dreplacement_of_old_operators.cpp28 DepthFirstGraphNodeIterator graph_it(graph_); in run()
H A Dfrozen_linear_transpose.cpp30 DepthFirstGraphNodeIterator graph_it(graph_); in run()
H A Dcheck_strict_fusion.cpp116 DepthFirstGraphNodeIterator it(graph); in CheckStrictFusion()
H A Dbatch_mm.cpp467 DepthFirstGraphNodeIterator it(graph); in hasMMOperators()
H A Dconcat_opt.cpp588 DepthFirstGraphNodeIterator graph_it(graph_); in collectOptimizableConcats()
H A Dsymbolic_shape_runtime_fusion.cpp307 DepthFirstGraphNodeIterator it(graph); in inlineFallbackGraphAndAddSRCopyOutOp()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/dbr_quantization/
H A Dremove_redundant_aliases.cpp20 DepthFirstGraphNodeIterator it(g); in DBRQuantRemoveRedundantAliasesImpl()
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dexit_transforms.cpp645 DepthFirstGraphNodeIterator it(graph); in convertEnterExitNodesToWithBlocks()
719 DepthFirstGraphNodeIterator it(graph); in convertWithBlocksToEnterExitNodes()
H A Dir_emitter.cpp5723 DepthFirstGraphNodeIterator it(graph); in eraseListLiterals()
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_utils.cc229 DepthFirstGraphNodeIterator it(graph); in getNodeWithKind()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/interpreter/
H A Dcode_impl.h966 DepthFirstGraphNodeIterator graph_it(graph_); in process_ops_for_mobile()