| 634 | } |
| 635 | |
| 636 | void setLayerDeviceTypes( |
| 637 | INetworkDefinition const& network, IBuilderConfig& config, LayerDeviceTypes const& layerDeviceTypes) |
| 638 | { |
| 639 | for (int32_t layerIdx = 0; layerIdx < network.getNbLayers(); ++layerIdx) |
| 640 | { |
| 641 | auto* layer = network.getLayer(layerIdx); |
| 642 | auto const layerName = layer->getName(); |
| 643 | if (layerDeviceTypes.find(layerName) != layerDeviceTypes.end()) |
| 644 | { |
| 645 | DeviceType const deviceType = layerDeviceTypes.at(layerName); |
| 646 | config.setDeviceType(layer, deviceType); |
| 647 | } |
| 648 | } |
| 649 | } |
| 650 | |
| 651 | void setMemoryPoolLimits(IBuilderConfig& config, BuildOptions const& build) |
| 652 | { |
no test coverage detected