| 68 | |
| 69 | template <DataType T> |
| 70 | Status PReLU::UploadParameters( |
| 71 | const ::tflite::gpu::Tensor<Linear, T>& parameters, CLContext* context) { |
| 72 | LinearStorageCreateInfo create_info; |
| 73 | create_info.storage_type = |
| 74 | DeduceLinearStorageType(definition_.GetPrimaryStorageType()); |
| 75 | create_info.data_type = definition_.GetPrimaryDataType(); |
| 76 | RETURN_IF_ERROR( |
| 77 | CreateLinearStorage(create_info, parameters, context, &alpha_)); |
| 78 | return OkStatus(); |
| 79 | } |
| 80 | |
| 81 | } // namespace cl |
| 82 | } // namespace gpu |
no test coverage detected