Home
last modified time | relevance | path

Searched full:linear_dynamic (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h1085 std::string linear_dynamic = R"( in dynamic_quantized_linear_pattern_and_replacements() local
1098 %r = quantized::linear_dynamic(%a, %packed_params, %reduce_range) in dynamic_quantized_linear_pattern_and_replacements()
1102 {"quantized::linear_dynamic", in dynamic_quantized_linear_pattern_and_replacements()
1103 std::move(linear_dynamic), in dynamic_quantized_linear_pattern_and_replacements()
1110 std::string linear_dynamic = R"( in dynamic_quant_fusion_pattern_and_replacements() local
1122 %r = quantized::linear_dynamic(%a, %packed_params, %reduce_range) in dynamic_quant_fusion_pattern_and_replacements()
1137 {"quantized::linear_dynamic", in dynamic_quant_fusion_pattern_and_replacements()
1138 std::move(linear_dynamic), in dynamic_quant_fusion_pattern_and_replacements()
/aosp_15_r20/external/pytorch/test/quantization/pt2e/
H A Dtest_metadata_porting.py283 _tag_partitions(backend_string, "linear_dynamic", annotated_partitions)
345 _tag_partitions(backend_string, "linear_dynamic", annotated_partitions)
404 _tag_partitions(backend_string, "linear_dynamic", annotated_partitions)
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
H A Dlinear.py53 Y = torch.ops.quantized.linear_dynamic(
57 Y = torch.ops.quantized.linear_dynamic(
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py1607 FileCheck().check("quantized::linear_dynamic").run(model.graph)
3170 M(), x, "quantized::linear_dynamic", tracing=tracing, dynamic=True
3190 M(), x, "quantized::linear_dynamic", tracing=tracing, dynamic=True
3226 "quantized::linear_dynamic(", count, exactly=True
3265 M(), data, "quantized::linear_dynamic", tracing=tracing, dynamic=True
3268 "quantized::linear_dynamic(", 2, exactly=True
3386 "quantized::linear_dynamic",
3398 "quantized::linear_dynamic",
H A Dtest_ondevice_quantization.py327 if "quantized::linear_dynamic" in n.kind():
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml692 - quantized::linear_dynamic
1041 quantized::linear_dynamic: 21
H A Dmodel_ops.yaml434 quantized::linear_dynamic: 91
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear_dynamic.cpp794 TORCH_SELECTIVE_NAME("quantized::linear_dynamic"), in TORCH_LIBRARY_IMPL()
819 TORCH_SELECTIVE_NAME("_quantized::linear_dynamic"), in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/ao/sparsity/
H A Dtest_kernels.py59 dense_qlinear_dynamic = torch.ops.quantized.linear_dynamic
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A Dlibrary.cpp150 …m.def(TORCH_SELECTIVE_SCHEMA("quantized::linear_dynamic(Tensor X, __torch__.torch.classes.quantize… in TORCH_LIBRARY()
246 …m.def(TORCH_SELECTIVE_SCHEMA("_quantized::linear_dynamic(Tensor X, __torch__.torch.classes.quantiz… in TORCH_LIBRARY()
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A D_lower_to_native_backend.py366 torch.ops.quantized.linear_dynamic,
952 quantize_per_tensor_dynamic - dequantize - functional linear --> linear_dynamic
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py2860 result = torch.ops.quantized.linear_dynamic(X, w_packed)
3134 qlinear_dynamic = torch.ops.quantized.linear_dynamic
3227 msg="torch.ops.quantized.linear_dynamic results are off")
4288 torch.ops.quantized.linear_dynamic(dummy_input, W_prepack)
4289 torch.ops.quantized.linear_dynamic(dummy_input, W_prepack)
H A Dtest_quantized_module.py1624 Z_ref = torch.ops.quantized.linear_dynamic(X, W_pack, reduce_range=True)
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py1138 None if is_reference else ns.call_function(torch.ops.quantized.linear_dynamic),
6769 QuantType.DYNAMIC: ns.call_function(torch.ops.quantized.linear_dynamic),