Home
last modified time | relevance | path

Searched full:hardsigmoid_ (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/ops/
H A DMetalNeurons.mm70 static Tensor& hardsigmoid_(Tensor& input) {
86 m.impl(TORCH_SELECTIVE_NAME("aten::hardsigmoid_"), TORCH_FN(hardsigmoid_));
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DClamp.cpp328 Tensor& hardsigmoid_(Tensor& self) { in hardsigmoid_() function
329 return ops::activation_(self, VK_KERNEL(hardsigmoid_)); in hardsigmoid_()
602 m.impl(TORCH_SELECTIVE_NAME("aten::hardsigmoid_"), hardsigmoid_); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/torch/ao/ns/fx/
H A Dmappings.py243 "hardsigmoid_",
702 "hardsigmoid_",
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml277 - aten::hardsigmoid_
836 aten::hardsigmoid_: 2
H A Dtensor_ops.py84 a.hardsigmoid_(),
H A Dmodel_ops.yaml173 aten::hardsigmoid_: 2
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h898 auto hardsigmoid_ = in quant_fusion_pattern_and_replacements() local
899 getFixedQParamOpFusionInfo("aten::hardsigmoid_", {}, false); in quant_fusion_pattern_and_replacements()
1075 std::move(hardsigmoid_), in quant_fusion_pattern_and_replacements()
H A Dhelper.cpp202 {Symbol::aten("hardsigmoid_"), _per_tensor_asym_qparam},
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dqconfig_mapping.py48 "hardsigmoid_": default_fixed_qparams_range_0to1_observer,
H A Dutils.py98 "hardsigmoid_",
/aosp_15_r20/external/pytorch/torch/_decomp/
H A D__init__.py319 aten.hardsigmoid_,
H A Ddecompositions.py5097 register_inplace(aten.hardsigmoid_, aten.hardsigmoid)
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/
H A DMPSCNNTests.mm286 auto Y1 = at::hardsigmoid_(X);
287 auto Y2 = at::hardsigmoid_(X2).cpu();
/aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/
H A D_common_operator_config_utils.py121 "hardsigmoid_": _FIXED_QPARAM_OP_0TO1_CONSTRAINTS,
/aosp_15_r20/external/executorch/backends/qualcomm/quantizer/
H A Dannotators.py287 [torch.ops.aten.hardsigmoid.default, torch.ops.aten.hardsigmoid_.default]
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/
H A Dfunctional.py655 return torch._C._nn.hardsigmoid_(input) # type: ignore[attr-defined]
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A D_lower_to_native_backend.py62 "hardsigmoid_",
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_ops_to_mkldnn.cpp1156 aten::hardsigmoid_, in ConvertFrozenOpsToMKLDNN()
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py1018 "torch._C._nn.hardsigmoid_",
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp2657 TEST_F(VulkanAPITest, hardsigmoid_) { in TEST_F() argument
2661 at::hardsigmoid_(cpu); in TEST_F()
2662 at::hardsigmoid_(vulkan); in TEST_F()
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py2937 x.hardsigmoid_()
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py2305 return torch._C._nn.hardsigmoid_(input)
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops.cpp6200 torch::Tensor output = torch::hardsigmoid_(input); in TEST_F()
6201 torch::Tensor lazy_output = torch::hardsigmoid_(lazy_input); in TEST_F()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml11752 - func: hardsigmoid_(Tensor(a!) self) -> Tensor(a!)