Searched full:hardsigmoid_ (Results 1 – 24 of 24) sorted by relevance
| /external/pytorch/aten/src/ATen/native/metal/ops/ |
| D | MetalNeurons.mm | 70 static Tensor& hardsigmoid_(Tensor& input) {
86 m.impl(TORCH_SELECTIVE_NAME("aten::hardsigmoid_"), TORCH_FN(hardsigmoid_));
|
| /external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| 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()
|
| /external/pytorch/torch/ao/ns/fx/ |
| D | mappings.py | 243 "hardsigmoid_",
702 "hardsigmoid_",
|
| /external/pytorch/torch/ao/quantization/ |
| D | qconfig_mapping.py | 48 "hardsigmoid_": default_fixed_qparams_range_0to1_observer,
|
| D | utils.py | 98 "hardsigmoid_",
|
| /external/pytorch/test/mobile/model_test/ |
| D | tensor_ops.py | 84 a.hardsigmoid_(),
|
| D | model_ops.yaml | 173 aten::hardsigmoid_: 2
|
| D | coverage.yaml | 277 - aten::hardsigmoid_
836 aten::hardsigmoid_: 2
|
| /external/pytorch/torch/_decomp/ |
| D | __init__.py | 319 aten.hardsigmoid_,
|
| D | decompositions.py | 5097 register_inplace(aten.hardsigmoid_, aten.hardsigmoid)
|
| /external/pytorch/torch/csrc/jit/passes/quantization/ |
| 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()
|
| D | helper.cpp | 202 {Symbol::aten("hardsigmoid_"), _per_tensor_asym_qparam},
|
| /external/pytorch/torch/ao/quantization/backend_config/ |
| D | _common_operator_config_utils.py | 121 "hardsigmoid_": _FIXED_QPARAM_OP_0TO1_CONSTRAINTS,
|
| /external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/ |
| D | MPSCNNTests.mm | 286 auto Y1 = at::hardsigmoid_(X);
287 auto Y2 = at::hardsigmoid_(X2).cpu();
|
| /external/executorch/backends/qualcomm/quantizer/ |
| D | annotators.py | 287 [torch.ops.aten.hardsigmoid.default, torch.ops.aten.hardsigmoid_.default]
|
| /external/pytorch/torch/ao/nn/quantized/ |
| D | functional.py | 655 return torch._C._nn.hardsigmoid_(input) # type: ignore[attr-defined]
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | frozen_ops_to_mkldnn.cpp | 1156 aten::hardsigmoid_, in ConvertFrozenOpsToMKLDNN()
|
| /external/pytorch/torch/ao/quantization/fx/ |
| D | _lower_to_native_backend.py | 62 "hardsigmoid_",
|
| /external/pytorch/torch/_dynamo/ |
| D | trace_rules.py | 1018 "torch._C._nn.hardsigmoid_",
|
| /external/pytorch/aten/src/ATen/test/ |
| 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()
|
| /external/pytorch/test/quantization/jit/ |
| D | test_quantize_jit.py | 2937 x.hardsigmoid_()
|
| /external/pytorch/torch/nn/ |
| D | functional.py | 2305 return torch._C._nn.hardsigmoid_(input)
|
| /external/pytorch/test/cpp/lazy/ |
| 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()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | native_functions.yaml | 11752 - func: hardsigmoid_(Tensor(a!) self) -> Tensor(a!)
|