Home
last modified time | relevance | path

Searched refs:blocks_to_visit (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dinsert_quant_dequant.cpp609 std::stack<Block*> blocks_to_visit; in ReplicateClampScalarArgs() local
612 blocks_to_visit.push(graph->block()); in ReplicateClampScalarArgs()
613 while (!blocks_to_visit.empty()) { in ReplicateClampScalarArgs()
614 Block* b = blocks_to_visit.top(); in ReplicateClampScalarArgs()
615 blocks_to_visit.pop(); in ReplicateClampScalarArgs()
623 blocks_to_visit.push(subblock); in ReplicateClampScalarArgs()
1113 std::stack<Block*> blocks_to_visit; in getInvokedMethods() local
1114 blocks_to_visit.push(graph->block()); in getInvokedMethods()
1115 while (!blocks_to_visit.empty()) { in getInvokedMethods()
1116 Block* b = blocks_to_visit.top(); in getInvokedMethods()
[all …]
H A Dinsert_observers.cpp921 std::stack<Block*> blocks_to_visit; in getInvokedMethods() local
922 blocks_to_visit.push(graph->block()); in getInvokedMethods()
923 while (!blocks_to_visit.empty()) { in getInvokedMethods()
924 Block* b = blocks_to_visit.top(); in getInvokedMethods()
925 blocks_to_visit.pop(); in getInvokedMethods()
939 blocks_to_visit.push(subblock); in getInvokedMethods()
1079 std::stack<Block*> blocks_to_visit; in fillPassThroughValueMap() local
1080 blocks_to_visit.push(graph->block()); in fillPassThroughValueMap()
1081 while (!blocks_to_visit.empty()) { in fillPassThroughValueMap()
1082 Block* b = blocks_to_visit.top(); in fillPassThroughValueMap()
[all …]
H A Ddedup_module_uses.cpp30 std::stack<Block*> blocks_to_visit; in findModuleUses() local
31 blocks_to_visit.push(graph->block()); in findModuleUses()
33 while (!blocks_to_visit.empty()) { in findModuleUses()
34 Block* b = blocks_to_visit.top(); in findModuleUses()
35 blocks_to_visit.pop(); in findModuleUses()
38 blocks_to_visit.push(subblock); in findModuleUses()
H A Dregister_packed_params.cpp61 std::stack<Block*> blocks_to_visit; in RegisterPrePackParams() local
63 blocks_to_visit.push(graph->block()); in RegisterPrePackParams()
68 while (!blocks_to_visit.empty()) { in RegisterPrePackParams()
69 Block* b = blocks_to_visit.top(); in RegisterPrePackParams()
70 blocks_to_visit.pop(); in RegisterPrePackParams()
141 blocks_to_visit.push(subblock); in RegisterPrePackParams()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dprepack_folding.cpp18 std::stack<Block*> blocks_to_visit; in PrePackingOpsFolder() local
20 blocks_to_visit.push(graph->block()); in PrePackingOpsFolder()
23 while (!blocks_to_visit.empty()) { in PrePackingOpsFolder()
24 Block* b = blocks_to_visit.top(); in PrePackingOpsFolder()
25 blocks_to_visit.pop(); in PrePackingOpsFolder()
53 blocks_to_visit.push(subblock); in PrePackingOpsFolder()
H A Dhoist_conv_packed_params.cpp86 std::stack<Block*> blocks_to_visit; in HoistConvPackedParams() local
87 blocks_to_visit.push(graph->block()); in HoistConvPackedParams()
92 while (!blocks_to_visit.empty()) { in HoistConvPackedParams()
93 Block* b = blocks_to_visit.top(); in HoistConvPackedParams()
94 blocks_to_visit.pop(); in HoistConvPackedParams()
131 blocks_to_visit.push(subblock); in HoistConvPackedParams()
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_cs_debug_info_serialization.cpp106 std::stack<Block*> blocks_to_visit; in TEST() local
114 blocks_to_visit.push(graph->block()); in TEST()
115 while (!blocks_to_visit.empty()) { in TEST()
116 Block* b = blocks_to_visit.top(); in TEST()
117 blocks_to_visit.pop(); in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/jit/ir/
H A Dsubgraph_matcher.cpp349 std::stack<Block*> blocks_to_visit; in findPatternMatches() local
353 blocks_to_visit.push(graph.block()); in findPatternMatches()
354 while (!blocks_to_visit.empty()) { in findPatternMatches()
355 Block* block = blocks_to_visit.top(); in findPatternMatches()
356 blocks_to_visit.pop(); in findPatternMatches()
362 blocks_to_visit.push(subblock); in findPatternMatches()
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/
H A Dbackend_detail.cpp45 std::stack<Block*> blocks_to_visit; in generate_debug_handles() local
49 blocks_to_visit.push(graph->block()); in generate_debug_handles()
50 while (!blocks_to_visit.empty()) { in generate_debug_handles()
51 Block* b = blocks_to_visit.top(); in generate_debug_handles()
52 blocks_to_visit.pop(); in generate_debug_handles()
57 blocks_to_visit.push(subblock); in generate_debug_handles()