/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/ops/ |
H A D | MetalNeurons.mm | 70 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 D | Clamp.cpp | 328 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 D | mappings.py | 243 "hardsigmoid_", 702 "hardsigmoid_",
|
/aosp_15_r20/external/pytorch/test/mobile/model_test/ |
H A D | coverage.yaml | 277 - aten::hardsigmoid_ 836 aten::hardsigmoid_: 2
|
H A D | tensor_ops.py | 84 a.hardsigmoid_(),
|
H A D | model_ops.yaml | 173 aten::hardsigmoid_: 2
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
H A D | quantization_patterns.h | 898 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 D | helper.cpp | 202 {Symbol::aten("hardsigmoid_"), _per_tensor_asym_qparam},
|
/aosp_15_r20/external/pytorch/torch/ao/quantization/ |
H A D | qconfig_mapping.py | 48 "hardsigmoid_": default_fixed_qparams_range_0to1_observer,
|
H A D | utils.py | 98 "hardsigmoid_",
|
/aosp_15_r20/external/pytorch/torch/_decomp/ |
H A D | __init__.py | 319 aten.hardsigmoid_,
|
H A D | decompositions.py | 5097 register_inplace(aten.hardsigmoid_, aten.hardsigmoid)
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/ |
H A D | MPSCNNTests.mm | 286 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.py | 121 "hardsigmoid_": _FIXED_QPARAM_OP_0TO1_CONSTRAINTS,
|
/aosp_15_r20/external/executorch/backends/qualcomm/quantizer/ |
H A D | annotators.py | 287 [torch.ops.aten.hardsigmoid.default, torch.ops.aten.hardsigmoid_.default]
|
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/ |
H A D | functional.py | 655 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.py | 62 "hardsigmoid_",
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
H A D | frozen_ops_to_mkldnn.cpp | 1156 aten::hardsigmoid_, in ConvertFrozenOpsToMKLDNN()
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | trace_rules.py | 1018 "torch._C._nn.hardsigmoid_",
|
/aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
H A D | vulkan_api_test.cpp | 2657 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 D | test_quantize_jit.py | 2937 x.hardsigmoid_()
|
/aosp_15_r20/external/pytorch/torch/nn/ |
H A D | functional.py | 2305 return torch._C._nn.hardsigmoid_(input)
|
/aosp_15_r20/external/pytorch/test/cpp/lazy/ |
H A D | test_lazy_ops.cpp | 6200 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 D | native_functions.yaml | 11752 - func: hardsigmoid_(Tensor(a!) self) -> Tensor(a!)
|