Home
last modified time | relevance | path

Searched defs:match_vmap (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_subgraph_rewriter.cpp35 const auto& match_vmap = match.values_map; in TEST() local
42 const auto& match_vmap = match.values_map; in TEST() local
49 const auto& match_vmap = match.values_map; in TEST() local
106 const auto& match_vmap = match.values_map; in TEST() local
260 const auto& match_vmap = match.values_map; in TEST() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dgraph_rewrite_helper.cpp23 const std::unordered_map<const Value*, Value*>& match_vmap, in getValue() argument
30 const std::unordered_map<const Value*, Value*>& match_vmap, in getIValue() argument
39 const auto& match_vmap = match.values_map; in getConvParams() local
268 const auto& match_vmap = match.values_map; in isClampFusable() local
H A Dfuse_linear.cpp29 const auto& match_vmap = match.values_map; in FuseLinear() local
124 const auto& match_vmap = match.values_map; in SwapFunctionalLinear() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dhelper.cpp668 const auto& match_vmap = match.values_map; in is_int_constant() local
678 const auto& match_vmap = match.values_map; in is_functional() local
702 const auto& match_vmap = match.values_map; in is_module() local
H A Dinsert_quant_dequant.cpp566 const auto& match_vmap = match.values_map; in RemoveRedundantDequantize() local
593 const auto& match_vmap = match.values_map; in RemoveRedundantQuantizationOps() local
H A Dquantization_patterns.h236 const auto& match_vmap = match.values_map; in input_b_is_scalar() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dunpack_quantized_weights.cpp270 auto match_vmap = match.values_map; in unpackQuantizedWeightsHelper() local
725 auto match_vmap = match.values_map; in insertPermutesHelper() local