Home
last modified time | relevance | path

Searched full:linear_dynamic_fp16 (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h1125 std::string linear_dynamic_fp16 = R"( in dynamic_quant_fusion_pattern_and_replacements() local
1133 %r = quantized::linear_dynamic_fp16(%a, %packed_params) in dynamic_quant_fusion_pattern_and_replacements()
1140 {"quantized::linear_dynamic_fp16", in dynamic_quant_fusion_pattern_and_replacements()
1141 std::move(linear_dynamic_fp16), in dynamic_quant_fusion_pattern_and_replacements()
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
H A Dlinear.py61 Y = torch.ops.quantized.linear_dynamic_fp16(
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml693 - quantized::linear_dynamic_fp16
1042 quantized::linear_dynamic_fp16: 18
H A Dmodel_ops.yaml435 quantized::linear_dynamic_fp16: 78
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A D_lower_to_native_backend.py370 torch.ops.quantized.linear_dynamic_fp16,
953 to(torch.float16) - dequantize - functional linear --> linear_dynamic_fp16
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear_dynamic.cpp800 TORCH_SELECTIVE_NAME("quantized::linear_dynamic_fp16"), in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A Dlibrary.cpp152 …m.def(TORCH_SELECTIVE_SCHEMA("quantized::linear_dynamic_fp16(Tensor X, __torch__.torch.classes.qua… in TORCH_LIBRARY()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dops.cpp2371 quantized::linear_dynamic_fp16,
2375 "quantized::linear_dynamic_fp16(Tensor X, __torch__.torch.classes." in __anon11f46a8b7502()
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py3360 FileCheck().check("quantized::linear_dynamic_fp16").check_not(
3875 "quantized::linear_dynamic_fp16",
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py1775 ns.call_function(torch.ops.quantized.linear_dynamic_fp16),
6877 qlinear_fun = ns.call_function(torch.ops.quantized.linear_dynamic_fp16)
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_runtime.cc2338 %output = quantized::linear_dynamic_fp16(%input, %packed_params) in TEST()
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py3332 qlinear_dynamic = torch.ops.quantized.linear_dynamic_fp16