Home
last modified time | relevance | path

Searched full:hardswish (Results 1 – 25 of 198) sorted by relevance

12345678

/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
Dhardswish.c42 … "failed to create Hardswish operator with %zu channels: number of channels must be non-zero", in pytorch_qnnp_create_hardswish_nc_q8()
49 … "failed to create Hardswish operator with %.7g input scale: scale must be finite and positive", in pytorch_qnnp_create_hardswish_nc_q8()
56 … "failed to create Hardswish operator with %.7g output scale: scale must be finite and positive", in pytorch_qnnp_create_hardswish_nc_q8()
63 "failed to create Hardswish operator with [%" PRIu8 ", %" PRIu8 in pytorch_qnnp_create_hardswish_nc_q8()
83 "failed to allocate 256 bytes for Hardswish lookup table"); in pytorch_qnnp_create_hardswish_nc_q8()
94 // hardswish, no min/max functions in C in pytorch_qnnp_create_hardswish_nc_q8()
123 pytorch_qnnp_operator_t hardswish, in pytorch_qnnp_setup_hardswish_nc_q8() argument
136 hardswish->batch_size = 0; in pytorch_qnnp_setup_hardswish_nc_q8()
140 hardswish->batch_size = batch_size; in pytorch_qnnp_setup_hardswish_nc_q8()
141 hardswish->input = input; in pytorch_qnnp_setup_hardswish_nc_q8()
[all …]
/external/executorch/backends/xnnpack/test/ops/
Dhardswish.py14 class Hardswish(torch.nn.Module): class in TestHardswish
17 self.hardswish = torch.nn.Hardswish()
20 return self.hardswish(x)
24 return torch.nn.functional.hardswish(x)
28 Tester(self.Hardswish(), inputs)
30 .check_count({"torch.ops.aten.hardswish.default": 1})
56 .check_count({"torch.ops.aten.hardswish.default": 1})
/external/executorch/backends/vulkan/runtime/graph/ops/glsl/
Dactivations.h9 float hardswish(float x) { in hardswish() function
19 vec4 hardswish(vec4 tex) { in hardswish() function
21 hardswish(tex.x), hardswish(tex.y), hardswish(tex.z), hardswish(tex.w)); in hardswish()
Dunary_op.yaml41 - NAME: hardswish
42 OPERATOR: hardswish(X)
/external/pytorch/benchmarks/operator_benchmark/pt/
Dhardswish_test.py8 Microbenchmarks for the hardswish operators.
12 # Configs for hardswish ops
34 ["Hardswish", nn.Hardswish],
Dqactivation_test.py97 ("functional.hardswish", qF.hardswish),
/external/pytorch/aten/src/ATen/native/metal/ops/
DMetalHardswish.mm25 [[MetalContext sharedInstance] specializedPipelineState:"hardswish"
48 static Tensor hardswish(const at::Tensor& input) {
58 [[MetalContext sharedInstance] specializedPipelineState:"hardswish"
82 m.impl(TORCH_SELECTIVE_NAME("aten::hardswish"), TORCH_FN(hardswish));
/external/XNNPACK/bench/
Dhardswish.cc53 state.SkipWithError("failed to create HardSwish operator"); in xnnpack_hardswish_f32()
62 state.SkipWithError("failed to setup HardSwish operator"); in xnnpack_hardswish_f32()
69 state.SkipWithError("failed to run HardSwish operator"); in xnnpack_hardswish_f32()
76 state.SkipWithError("failed to delete HardSwish operator"); in xnnpack_hardswish_f32()
118 state.SkipWithError("failed to create HardSwish operator"); in xnnpack_hardswish_f16()
127 state.SkipWithError("failed to setup HardSwish operator"); in xnnpack_hardswish_f16()
134 state.SkipWithError("failed to run HardSwish operator"); in xnnpack_hardswish_f16()
141 state.SkipWithError("failed to delete HardSwish operator"); in xnnpack_hardswish_f16()
209 builder.CreateString("HardSwish model"), in tflite_hardswish_f32()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dhard_swish_test.cc28 TEST(HardSwish, 4D) {
47 TEST(HardSwish, 3D) {
65 TEST(HardSwish, 2D) {
82 TEST(HardSwish, 1D) {
97 TEST(HardSwish, MultiThreading) { in TEST() argument
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/
Dhardswish.cc49 state.SkipWithError("failed to create Hardswish operator"); in hardswish_q8()
60 state.SkipWithError("failed to setup Hardswish operator"); in hardswish_q8()
67 state.SkipWithError("failed to run Hardswish operator"); in hardswish_q8()
81 state.SkipWithError("failed to delete Hardswish operator"); in hardswish_q8()
/external/pytorch/aten/src/ATen/native/quantized/cpu/
Dqhardswish.cpp57 "failed to create QNNPACK Hardswish operator"); in qnnpack_hardswish()
67 "failed to setup QNNPACK Hardswish operator"); in qnnpack_hardswish()
76 "failed to run QNNPACK Hardswish operator"); in qnnpack_hardswish()
103 m.impl(TORCH_SELECTIVE_NAME("quantized::hardswish"), TORCH_FN(quantized_hardswish)); in TORCH_LIBRARY_IMPL()
/external/pytorch/torch/onnx/
Dsymbolic_opset14.py8 HardSwish, Trilu
30 "hardswish",
42 @_onnx_symbolic("aten::hardswish")
44 def hardswish(g: jit_utils.GraphContext, self): function
45 return g.op("HardSwish", self)
122 @_onnx_symbolic("quantized::hardswish")
126 output = hardswish(g, x)
/external/pytorch/torch/ao/ns/fx/
Dmappings.py172 # Hardswish
174 nn.Hardswish,
203 # F.hardswish
205 F.hardswish,
495 F.hardswish,
523 toq.hardswish,
606 nn.Hardswish,
656 nnq.Hardswish,
/external/pytorch/torch/ao/nn/quantized/modules/
Dactivation.py9 "Hardswish",
59 class Hardswish(torch.nn.Hardswish): class
60 r"""This is the quantized version of :class:`~torch.nn.Hardswish`.
74 return torch.ops.quantized.hardswish(input, self.scale, self.zero_point)
82 return Hardswish(float(scale), int(zero_point))
/external/executorch/backends/xnnpack/partition/
Dconfigs.py50 exir_ops.edge.aten.hardswish.default,
70 # TODO(T161981984) recomposed hardswish into a single node
71 torch.nn.Hardswish, # we need to recompose
119 torch.nn.Hardswish,
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/
DCMakeLists.txt165 src/hardswish.c
557 add_executable(hardswish-test test/hardswish.cc)
558 set_target_properties(hardswish-test PROPERTIES
562 target_include_directories(hardswish-test PRIVATE src test)
563 target_link_libraries(hardswish-test PRIVATE pytorch_qnnpack cpuinfo gtest gtest_main)
564 add_test(hardswish-test hardswish-test)
810 add_executable(hardswish-bench bench/hardswish.cc)
811 set_target_properties(hardswish-bench PROPERTIES
815 target_link_libraries(hardswish-bench PRIVATE pytorch_qnnpack benchmark)
Dconfigure.py90 build.cc("hardswish.c"),
226 build.unittest("hardswish-test", build.cxx("hardswish.cc"))
265 build.benchmark("hardswish-bench", build.cxx("hardswish.cc"))
/external/pytorch/torch/nn/quantized/modules/
D__init__.py29 Hardswish,
78 "Hardswish",
/external/executorch/backends/vulkan/runtime/graph/ops/impl/
DUnaryOp.cpp138 DEFINE_ACTIVATION_FN(hardswish);
156 VK_REGISTER_OP(aten.hardswish.default, hardswish);
/external/pytorch/test/quantization/ao_migration/
Dtest_ao_migration.py24 "hardswish",
51 "Hardswish",
74 "Hardswish",
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_hardswish.cc28 // This transformation rule tries to identify the HardSwish structure generated
30 // The formula of hardswish is:
107 AddMessageF("Creating hardswish op (%s) replacing equivalent subgraph", in Run()
/external/armnn/src/backends/backendsCommon/test/
DActivationEndToEndTestImpl.hpp35 case ActivationFunction::HardSwish: in GetActivationTolerance()
145 /** Executes an end to end test for HardSwish activation with specific input and expected-output da…
164 armnn::ActivationDescriptor descriptor(ActivationFunction::HardSwish, 1.0); in HardSwishEndToEndTest()
/external/pytorch/torch/ao/quantization/
Dquantization_mappings.py92 nn.Hardswish: nnq.Hardswish,
183 F.hardswish: torch.ops.quantized.hardswish,
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dhardswish.py15 """Test configs for hardswish."""
44 """Make a set of tests to do hardswish."""
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dhardswish_op_builder.cc63 TF_LITE_KERNEL_LOG(context, "Wrong # of inputs to hardswish!."); in RegisterInputs()
73 TF_LITE_KERNEL_LOG(context, "Wrong # of outputs to hardswish!."); in RegisterOutputs()

12345678