25 auto prelu = [](
float x,
float alpha)
27 return x < 0 ? alpha * x : x;
30 BroadcastLoop(inputShape, alphaShape, outputShape).
Unroll(prelu, 0, inputData, alphaData, outputData);
const TensorShape & GetShape() const
Copyright (c) 2021 ARM Limited and Contributors.
void PreluImpl(const TensorInfo &inputInfo, const TensorInfo &alphaInfo, const TensorInfo &outputInfo, Decoder< float > &inputData, Decoder< float > &alphaData, Encoder< float > &outputData)
void Unroll(Func operationFunc, unsigned int dimension, DecoderOp &inData0, DecoderOp &inData1, EncoderOp &outData)