27 auto prelu = [](
float x,
float alpha)
29 return x < 0 ? alpha * x : x;
32 BroadcastLoop(inputShape, alphaShape, outputShape).
Unroll(prelu, 0, inputData, alphaData, outputData);
void Unroll(Func operationFunc, unsigned int dimension, DecoderOp &inData0, DecoderOp &inData1, EncoderOp &outData)
void PreluImpl(const PreluQueueDescriptor &data, Decoder< float > &inputData, Decoder< float > &alphaData, Encoder< float > &outputData)
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs
const TensorShape & GetShape() const
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers