| 33 | namespace cl { |
| 34 | |
| 35 | class PReLU : public ElementwiseOperation { |
| 36 | public: |
| 37 | PReLU() = default; |
| 38 | // Move only |
| 39 | PReLU(PReLU&& operation); |
| 40 | PReLU& operator=(PReLU&& operation); |
| 41 | PReLU(const PReLU&) = delete; |
| 42 | PReLU& operator=(const PReLU&) = delete; |
| 43 | |
| 44 | void SetLinkIndex(int index) override; |
| 45 | std::string GetCoreCode(const std::string& src, const std::string& z_coord, |
| 46 | const std::string& address) const override; |
| 47 | std::string GetArgsDeclaration() const override; |
| 48 | Status BindArguments(CLKernel* kernel) override; |
| 49 | |
| 50 | friend Status CreatePReLU(const CreationContext& creation_context, |
| 51 | const OperationDef& definition, |
| 52 | const PReLUAttributes& attr, PReLU* result); |
| 53 | |
| 54 | private: |
| 55 | PReLU(const OperationDef& definition, const PReLUAttributes& attr); |
| 56 | |
| 57 | template <DataType T> |
| 58 | Status UploadParameters(const ::tflite::gpu::Tensor<Linear, T>& parameters, |
| 59 | CLContext* context); |
| 60 | |
| 61 | FLT clip_; |
| 62 | LinearStorage alpha_; |
| 63 | }; |
| 64 | |
| 65 | Status CreatePReLU(const CreationContext& creation_context, |
| 66 | const OperationDef& definition, const PReLUAttributes& attr, |