/aosp_15_r20/external/pytorch/test/jit/ |
H A D | test_optimize_for_mobile_preserve_debug_info.py | 238 conv2d_activation=F.hardtanh_, 239 conv2d_activation_kind="aten::hardtanh_", 245 linear_activation=F.hardtanh_, 246 linear_activation_kind="aten::hardtanh_",
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/ops/ |
H A D | MetalClamp.mm | 13 static Tensor& hardtanh_(Tensor& input, const Scalar& min_val, const Scalar& max_val) { 62 m.impl(TORCH_SELECTIVE_NAME("aten::hardtanh_"), TORCH_FN(hardtanh_));
|
/aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
H A D | vulkan_api_test.cpp | 2714 TEST_F(VulkanAPITest, hardtanh_) { in TEST_F() argument 2718 at::hardtanh_(a_cpu, 3, 7); in TEST_F() 2719 at::hardtanh_(a_vulkan, 3, 7); in TEST_F() 7159 hardtanh_, enumerator 7238 class Hardtanh_ final : public BaseOp { class 7240 Hardtanh_() : BaseOp(OpType::hardtanh_) {} in Hardtanh_() function in __anon75b2796d0211::Hardtanh_ 7243 return at::hardtanh_(input, 0, 6); in run() 7247 return "hardtanh_"; in toString() 7301 ops_.emplace_back(new Hardtanh_()); in MobileNetV2() 7303 ops_.emplace_back(new Hardtanh_()); in MobileNetV2() [all …]
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
H A D | Clamp.cpp | 290 Tensor& hardtanh_(Tensor& self, const Scalar& min, const Scalar& max) { in hardtanh_() function 608 m.impl(TORCH_SELECTIVE_NAME("aten::hardtanh_"), hardtanh_); in TORCH_LIBRARY_IMPL()
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_xnnpack_integration.py | 1086 "aten::hardtanh_": 2, 1093 M(F.hardtanh_), pattern_count_map, data_shape, prepack_removal=True 1097 pattern_count_map["aten::hardtanh_"] = -1 1099 M(F.hardtanh_),
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
H A D | metal_rewrite.cpp | 189 %r = aten::hardtanh_(%r, %output_min, %output_max) in fuseHardtanhWithPackedOps() 197 %res = aten::hardtanh_(%linear_res, %output_min, %output_max) in fuseHardtanhWithPackedOps()
|
H A D | xnnpack_rewrite.cpp | 229 %res = aten::hardtanh_(%linear_res, %output_min, %output_max) in fuseHardtanhWithPackedOps() 239 %res = aten::hardtanh_(%conv2d_res, %output_min, %output_max) in fuseHardtanhWithPackedOps()
|
/aosp_15_r20/external/pytorch/torch/ao/ns/fx/ |
H A D | mappings.py | 252 F.hardtanh_, 548 F.hardtanh_,
|
/aosp_15_r20/external/pytorch/ios/TestApp/custom_build/ |
H A D | mobilenetv2.yaml | 6 - aten::hardtanh_
|
/aosp_15_r20/external/pytorch/tools/pyi/ |
H A D | gen_pyi.py | 390 "hardtanh_": [ 391 "def hardtanh_({}) -> Tensor: ...".format( 516 "hardtanh_",
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
H A D | helper.cpp | 162 "hardtanh_", 169 "hardtanh_",
|
H A D | quantization_patterns.h | 887 auto hardtanh_ = getClampOpFusionInfo("aten::hardtanh_", {"%min", "%max"}); in quant_fusion_pattern_and_replacements() local 1070 std::move(hardtanh_), in quant_fusion_pattern_and_replacements()
|
/aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/ |
H A D | graph_utils.py | 27 {torch.nn.Hardtanh, torch.nn.functional.hardtanh, torch.nn.functional.hardtanh_},
|
/aosp_15_r20/external/pytorch/docs/source/ |
H A D | nn.functional.rst | 77 hardtanh_
|
/aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
H A D | test_x86inductor_quantizer.py | 629 torch.ops.aten.hardtanh_.default, 634 torch.ops.aten.hardtanh_.default, 1762 torch.ops.aten.hardtanh_.default, 1767 torch.ops.aten.hardtanh_.default,
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | Activation.cpp | 457 Tensor& hardtanh_(Tensor& self, const Scalar& min, const Scalar& max) { in hardtanh_() function 534 return at::hardtanh_(self, /*min_val=*/0, /*max_val=*/6); in relu6_()
|
/aosp_15_r20/external/executorch/backends/arm/quantizer/ |
H A D | arm_quantizer_utils.py | 141 torch.ops.aten.hardtanh_.default,
|
/aosp_15_r20/external/pytorch/test/mobile/model_test/ |
H A D | coverage.yaml | 281 - aten::hardtanh_ 839 aten::hardtanh_: 3
|
H A D | tensor_ops.py | 88 a.hardtanh_(),
|
H A D | model_ops.yaml | 176 aten::hardtanh_: 5
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
H A D | interface.cpp | 57 aten::hardtanh_, in fuseGraph()
|
/aosp_15_r20/external/pytorch/torch/_decomp/ |
H A D | __init__.py | 324 aten.hardtanh_,
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/ |
H A D | MPSCNNTests.mm | 911 auto Y1 = at::hardtanh_(X1, 0, 6.0); 913 auto Y2 = at::hardtanh_(X2, 0, 6.0).cpu();
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/ |
H A D | TracerRunner.cpp | 38 "aten::hardtanh_",
|
/aosp_15_r20/external/pytorch/test/quantization/jit/ |
H A D | test_quantize_jit.py | 2743 self.hardtanh_ = torch.nn.Hardtanh(inplace=True) 2754 self.hardtanh_(x) 2756 F.hardtanh_(x) 2761 ["aten::clamp", "aten::hardtanh", "aten::hardtanh_"], [True, False]
|