Searched refs:op_pattern (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 110 std::string op_pattern = "graph(%a_quant" + extra_op_arg_list + "):" + R"( in getInputTensorQParamOpPattern() local
119 return op_pattern; in getInputTensorQParamOpPattern()
126 std::string op_pattern = in getInputTensorQParamOpFusionInfo() local
133 return {op_name, std::move(op_pattern), std::move(op_replacement)}; in getInputTensorQParamOpFusionInfo()
143 std::string op_pattern = variable
151 return {op_name, std::move(op_pattern), std::move(op_replacement), filters};
169 std::string op_pattern = graph_header; in getClampOpFusionInfo() local
171 op_pattern += getQuantizeForScalar(arg); in getClampOpFusionInfo()
172 op_pattern += getDequantize(arg); in getClampOpFusionInfo()
173 op_pattern += getItem(arg); in getClampOpFusionInfo()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | fuser_method_mappings.py | 245 def _get_valid_patterns(op_pattern): argument
268 if isinstance(op_pattern, (tuple, list)):
270 for sub_pattern in op_pattern:
274 result = [op_pattern, MatchAllNode]
279 op_pattern: Pattern,
287 op_patterns = _get_valid_patterns(op_pattern)
289 for op_pattern in op_patterns:
290 fuser_method = fuser_method_mapping.get(op_pattern, None)
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | quantize_nodes.cc | 644 string op_pattern; in QuantizeNodes() local
649 strings::StrAppend(&op_pattern, (is_first ? "" : "|"), in QuantizeNodes()
694 converted_graph_def, {op_pattern}, in QuantizeNodes()
|