| /aosp_15_r20/external/pytorch/torch/_export/passes/ |
| H A D | replace_quantized_ops_with_standard_ops_pass.py | 263 if opname in ["conv1d_relu", "conv2d_relu", "linear_relu", "add_relu", "mul_relu"]:
302 if opname in ["conv2d", "conv2d_relu"]
476 "conv2d_relu.new": _transform_conv_with_packedparam,
478 "conv2d_relu.default": _transform_conv_with_packedparam,
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | unpack_quantized_weights.cpp | 69 } else if (input_name == "quantized::conv2d_relu") { in getScaleFromInput()
70 // %r = quantized::conv2d_relu(%input, %packed_weight, %w_scale, in getScaleFromInput()
74 "quantized::conv2d_relu expected scale to be 3rd input"); in getScaleFromInput()
616 %r = quantized::conv2d_relu(%input, %packed_params, %scale, %zero_point) in UnpackQuantizedWeights()
755 …%r = quantized::conv2d_relu(%input, %weight, %bias, %stride, %padding, %dilation, %groups, %w_scal… in insertPermutes()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 341 std::string conv2d_relu = R"( in quant_fusion_pattern_and_replacements() local
368 // quantized::conv2d_relu in quant_fusion_pattern_and_replacements()
371 %r_quant = quantized::conv2d_relu(%a_quant, %packed_params, %r_scale, %r_zero_point) in quant_fusion_pattern_and_replacements()
947 {"quantized::conv2d_relu", std::move(conv2d_relu), quantized_conv2d_relu}, in quant_fusion_pattern_and_replacements()
948 {"quantized::conv2d_relu", in quant_fusion_pattern_and_replacements()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | import.cpp | 197 …%r = quantized::conv2d_relu(%x, %packed_params, %stride, %padding, %dilation, %groups, %r_scale, %… in rewriteQuantizedConvForBC()
217 %r = quantized::conv2d_relu(%x, %packed_params, %r_scale, %r_zero_point) in rewriteQuantizedConvForBC()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/ |
| H A D | library.cpp | 64 …m.def(TORCH_SELECTIVE_SCHEMA("quantized::conv2d_relu.new(Tensor qx, __torch__.torch.classes.quanti… in TORCH_LIBRARY()
70 …m.def(TORCH_SELECTIVE_SCHEMA("quantized::conv2d_relu(Tensor qx, __torch__.torch.classes.quantized.… in TORCH_LIBRARY()
235 …m.def(TORCH_SELECTIVE_SCHEMA("_quantized::conv2d_relu(Tensor qx, __torch__.torch.classes.quantized… in TORCH_LIBRARY()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | vulkan_rewrite.cpp | 257 // quantized::conv2d_relu in rewriteQuantizedOps()
260 %res = quantized::conv2d_relu(%a_quant, %packed_params, %r_scale, %r_zero_point) in rewriteQuantizedOps()
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_equalize_fx.py | 749 ns.call_function(torch.ops.quantized.conv2d_relu),
756 ns.call_function(torch.ops.quantized.conv2d_relu),
|
| H A D | test_quantize_fx.py | 7031 2: ns.call_function(torch.ops.quantized.conv2d_relu),
7036 2: ns.call_function(torch.ops.quantized.conv2d_relu),
7091 """tests for conv1d_relu/conv2d_relu/conv3d_relu"""
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 674 - quantized::conv2d_relu.new
1033 quantized::conv2d_relu.new: 50
|
| H A D | model_ops.yaml | 425 quantized::conv2d_relu.new: 83
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/ |
| H A D | conv_relu.py | 152 return torch.ops.quantized.conv2d_relu(
|
| /aosp_15_r20/external/pytorch/torch/ao/ns/fx/ |
| H A D | weight_utils.py | 209 toq.conv2d_relu: get_qconv_fun_weight,
|
| H A D | mappings.py | 518 toq.conv2d_relu,
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_caffe2.py | 122 def conv2d_relu( function
|
| H A D | symbolic_opset10.py | 975 @_onnx_symbolic("quantized::conv2d_relu")
|
| H A D | symbolic_opset13.py | 905 @_onnx_symbolic("quantized::conv2d_relu")
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv.cpp | 1976 m.impl(TORCH_SELECTIVE_NAME("quantized::conv2d_relu.new"), QConvInt8<2, true>::run); in TORCH_LIBRARY_IMPL()
1983 m.impl(TORCH_SELECTIVE_NAME("quantized::conv2d_relu"), QConvInt8ForBC<2, true>::run); in TORCH_LIBRARY_IMPL()
1997 m.impl(TORCH_SELECTIVE_NAME("_quantized::conv2d_relu"), QConvInt8<2, true>::run); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | Conv.cpp | 402 m.impl(TORCH_SELECTIVE_NAME("quantized::conv2d_relu.new"), QConvInt8<2, true>::run); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | _lower_to_native_backend.py | 339 F.conv2d: (torch.ops.quantized.conv2d, torch.ops.quantized.conv2d_relu),
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | conf.py | 1246 "conv2d_relu",
|
| /aosp_15_r20/external/pytorch/torch/backends/_nnapi/ |
| H A D | serializer.py | 894 "quantized::conv2d_relu": lambda self, node: self.add_qconv2d(
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | lowerings.cpp | 1953 …{"quantized::conv2d_relu.new(Tensor qx, __torch__.torch.classes.quantized.Conv2dPackedParamsBase p… in nnc_lowerings_lazy_registration()
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 5256 qconv = torch.ops.quantized.conv2d_relu
5546 qconv = torch.ops.quantized.conv2d_relu
|
| /aosp_15_r20/external/pytorch/test/quantization/jit/ |
| H A D | test_quantize_jit.py | 1708 """tests for conv1d_relu/conv2d_relu/conv3d_relu"""
|