Searched defs:hardtanh (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | activation.h | 107 inline Tensor hardtanh( in hardtanh() function
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/ |
| H A D | functional.py | 592 def hardtanh( function
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | quantization_patterns.h | 885 auto hardtanh = getClampOpFusionInfo("aten::hardtanh", {"%min", "%max"}); in quant_fusion_pattern_and_replacements() local
|
| /aosp_15_r20/external/pytorch/torch/_refs/nn/functional/ |
| H A D | __init__.py | 1012 def hardtanh( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Clamp.cpp | 286 Tensor hardtanh(const Tensor& self, const Scalar& min, const Scalar& max) { in hardtanh() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Activation.cpp | 433 Tensor hardtanh(const Tensor& self, const Scalar& min, const Scalar& max) { in hardtanh() function
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 95 def hardtanh(g: jit_utils.GraphContext, self: _C.Value, min_val: float, max_val: float): function
|
| H A D | symbolic_opset9.py | 3773 def hardtanh(g: jit_utils.GraphContext, self: _C.Value, min_val: float, max_val: float): function
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 1745 def hardtanh( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 2699 TEST_F(VulkanAPITest, hardtanh) { in TEST_F() argument
|