Home
last modified time | relevance | path

Searched defs:prelu (Results 1 – 12 of 12) sorted by relevance

/external/XNNPACK/test/
Dprelu-microkernel-tester.h95 void Test(xnn_f16_prelu_ukernel_function prelu) const { in Test()
141 void Test(xnn_f32_prelu_ukernel_function prelu) const { in Test()
/external/armnn/src/backends/reference/workloads/
DPreluImpl.cpp25 auto prelu = [](float x, float alpha) in PreluImpl() local
/external/armnn/src/backends/backendsCommon/test/
DPreluEndToEndTestImpl.hpp28 IConnectableLayer* prelu = net->AddPreluLayer("Prelu"); in CreatePreluNetwork() local
DActivationEndToEndTestImpl.hpp61 IConnectableLayer* prelu = net->AddActivationLayer(descriptor, ActivationName); in CreateActivationNetwork() local
/external/pytorch/torch/onnx/
Dsymbolic_opset8.py225 def prelu(g: jit_utils.GraphContext, self, weight): function
Dsymbolic_opset9.py1183 def prelu(g: jit_utils.GraphContext, self, weight): function
/external/pytorch/torch/csrc/api/include/torch/nn/functional/
Dactivation.h399 inline Tensor prelu(const Tensor& input, const Tensor& weight) { in prelu() function
/external/pytorch/torch/_refs/nn/functional/
D__init__.py1115 def prelu(a: TensorLikeType, weight: TensorLikeType) -> TensorLikeType: function
/external/XNNPACK/src/xnnpack/
Doperator.h287 struct prelu_context prelu; member
Dparams.h564 struct prelu_parameters prelu; member
617 struct prelu_parameters prelu; member
/external/pytorch/aten/src/ATen/native/
DActivation.cpp689 Tensor prelu(const Tensor& self, const Tensor& weight_) { in prelu() function
/external/pytorch/torch/nn/
Dfunctional.py1916 prelu = _add_docstr( variable