diff options
| author | David Beck <david.beck@arm.com> | 2018-10-18 15:13:56 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
| commit | 33f0ae0d293f5048089f2a04985a8a8bfa1d75a6 (patch) | |
| tree | 19f27463e524150c26ac75a2ea5c8e6c95f954c6 /src/armnn/Network.cpp | |
| parent | ceae3aa1b619161d49fd2847d3c73d6a858b2b8c (diff) | |
| download | armnn-33f0ae0d293f5048089f2a04985a8a8bfa1d75a6.tar.gz | |
IVGCVSW-2019 : replace Compute enum in the Layer object
Change-Id: I76551d511ef718eac36e5b8e5fe426ec3a402855
Diffstat (limited to 'src/armnn/Network.cpp')
| -rw-r--r-- | src/armnn/Network.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 51490e33c4..c43f336145 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -187,7 +187,7 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, { // need to set the compute device on the layer // before we can check if it is supported - layer->SetComputeDevice(backend); + layer->SetBackendId(backend); if (!IWorkloadFactory::IsLayerSupported(*layer, dataType, reasonIfUnsupported)) { if (dataType == DataType::Float16) @@ -211,7 +211,7 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, std::string reasonIfUnsupported; // Try preferred backend first - layer->SetComputeDevice(preferredBackend); + layer->SetBackendId(preferredBackend); if (IWorkloadFactory::IsLayerSupported(*layer, boost::none, reasonIfUnsupported)) { supportedBackendFound = true; @@ -226,7 +226,7 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, continue; } - layer->SetComputeDevice(backend); + layer->SetBackendId(backend); if (IWorkloadFactory::IsLayerSupported(*layer, boost::none, reasonIfUnsupported)) { supportedBackendFound = true; @@ -260,7 +260,7 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, } std::stringstream warningMsg; warningMsg << "WARNING: Layer of type " << GetLayerTypeAsCString(layer->GetType()) - << " is not supported on requested backend " << layer->GetComputeDevice() + << " is not supported on requested backend " << layer->GetBackendId().Get() << " for data type " << GetDataTypeName(dataType) << " (reason: " << reasonIfUnsupported << "), falling back to the next backend."; @@ -287,7 +287,7 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, layerType == armnn::LayerType::Constant || layerType == armnn::LayerType::Permute)) { - layer->SetComputeDevice(armnn::Compute::CpuRef); + layer->SetBackendId(armnn::Compute::CpuRef); } else { |