You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
1.9 KiB
72 lines
1.9 KiB
9 years ago
|
/* Copyright (c) 2016 Baidu, Inc. All Rights Reserve.
|
||
|
|
|
||
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
you may not use this file except in compliance with the License.
|
||
|
|
You may obtain a copy of the License at
|
||
|
|
|
||
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
|
||
|
|
Unless required by applicable law or agreed to in writing, software
|
||
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
See the License for the specific language governing permissions and
|
||
|
|
limitations under the License. */
|
||
|
|
|
||
|
|
|
||
|
|
#include "PaddleAPI.h"
|
||
|
|
#include "paddle/parameter/Parameter.h"
|
||
|
|
|
||
|
|
struct ParameterPrivate {
|
||
|
|
std::shared_ptr<paddle::Parameter> sharedPtr;
|
||
|
|
paddle::Parameter* rawPtr;
|
||
|
|
|
||
|
|
ParameterPrivate() : sharedPtr(nullptr), rawPtr(nullptr) {}
|
||
|
|
|
||
|
|
paddle::Parameter* getPtr() {
|
||
|
|
if (sharedPtr) {
|
||
|
|
return sharedPtr.get();
|
||
|
|
} else {
|
||
|
|
return rawPtr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
Parameter::Parameter() : m(new ParameterPrivate()) {}
|
||
|
|
|
||
|
|
Parameter::~Parameter() { delete m; }
|
||
|
|
|
||
|
|
Parameter* Parameter::createFromRawPtr(void* ptr) {
|
||
|
|
auto p = new Parameter();
|
||
|
|
p->m->rawPtr = *static_cast<paddle::Parameter**>(ptr);
|
||
|
|
return p;
|
||
|
|
}
|
||
|
|
|
||
|
|
Parameter* Parameter::createFromSharedPtr(void* ptr) {
|
||
|
|
auto& p = *(paddle::ParameterPtr*)(ptr);
|
||
|
|
if (p == nullptr) {
|
||
|
|
return nullptr;
|
||
|
|
} else {
|
||
|
|
auto retParam = new Parameter();
|
||
|
|
retParam->m->sharedPtr = p;
|
||
|
|
return retParam;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string Parameter::getName() const { return m->getPtr()->getName(); }
|
||
|
|
|
||
|
|
Vector* Parameter::getBuf(ParameterType type) {
|
||
|
|
auto buf = m->getPtr()->getBuf(type);
|
||
|
|
return Vector::createByPaddleVectorPtr(&buf);
|
||
|
|
}
|
||
|
|
|
||
|
|
ParameterConfig* Parameter::getConfig() {
|
||
|
|
if (m->sharedPtr) {
|
||
|
|
return ParameterConfig::createParameterConfigFromParameterSharedPtr(
|
||
|
|
&m->sharedPtr);
|
||
|
|
} else {
|
||
|
|
return ParameterConfig::createParameterConfigFromParameterPtr(m->rawPtr);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t Parameter::getID() const { return m->getPtr()->getID(); }
|