本文整理汇总了C++中ParameterPtr::getDeviceId方法的典型用法代码示例。如果您正苦于以下问题：C++ ParameterPtr::getDeviceId方法的具体用法？C++ ParameterPtr::getDeviceId怎么用？C++ ParameterPtr::getDeviceId使用的例子？那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类ParameterPtr的用法示例。
在下文中一共展示了ParameterPtr::getDeviceId方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: initSubNetwork
void RecurrentLayerGroup::initSubNetwork(
NeuralNetwork* rootNetwork,
const ModelConfig& config,
const std::vector<ParameterType>& parameterTypes,
bool useGpu) {
setNeedGradient(true);
network_.reset(new RecurrentGradientMachine(config_.name(), rootNetwork));
ParamInitCallback cb = [this, rootNetwork](int paramId, Parameter* para) {
para->enableSharedType(
PARAMETER_VALUE,
rootNetwork->getParameters()[paramId]->getBuf(PARAMETER_VALUE),
rootNetwork->getParameters()[paramId]->getMat(PARAMETER_VALUE));
para->enableSharedType(
PARAMETER_GRADIENT,
rootNetwork->getParameters()[paramId]->getBuf(PARAMETER_GRADIENT),
rootNetwork->getParameters()[paramId]->getMat(PARAMETER_GRADIENT));
};
network_->init(config, cb, parameterTypes, useGpu);
for (auto paramId : network_->getParameterIds()) {
ParameterPtr parameter = rootNetwork->getParameters()[paramId];
parameter->incShared();
CHECK_EQ(parameter->getDeviceId(), getDeviceId());
parameters_.push_back(parameter);
}
}