| 81 | } |
| 82 | |
| 83 | Status CreatePReLU(const CreationContext& creation_context, |
| 84 | const OperationDef& definition, const PReLUAttributes& attr, |
| 85 | PReLU* result) { |
| 86 | auto alpha = absl::get_if<::tflite::gpu::Tensor<Linear, DataType::FLOAT32>>( |
| 87 | &attr.alpha); |
| 88 | if (!alpha) { |
| 89 | return InvalidArgumentError("Alpha is missing"); |
| 90 | } |
| 91 | *result = PReLU(definition, attr); |
| 92 | RETURN_IF_ERROR(result->UploadParameters(*alpha, creation_context.context)); |
| 93 | result->SetLinkIndex(0); |
| 94 | return OkStatus(); |
| 95 | } |
| 96 | |
| 97 | } // namespace cl |
| 98 | } // namespace gpu |