Home
last modified time | relevance | path

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

/external/pytorch/test/cpp/jit/
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
/external/pytorch/torch/csrc/jit/passes/
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
Dfuse_linear.cpp29 const auto& match_vmap = match.values_map; in FuseLinear() local
124 const auto& match_vmap = match.values_map; in SwapFunctionalLinear() local
/external/pytorch/torch/csrc/jit/passes/quantization/
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
Dquantization_patterns.h236 const auto& match_vmap = match.values_map; in input_b_is_scalar() local
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
/external/pytorch/torch/csrc/jit/passes/onnx/
Dunpack_quantized_weights.cpp270 auto match_vmap = match.values_map; in unpackQuantizedWeightsHelper() local
725 auto match_vmap = match.values_map; in insertPermutesHelper() local