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/Graph.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/Graph.cpp')
-rw-r--r-- | src/armnn/Graph.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/armnn/Graph.cpp b/src/armnn/Graph.cpp index 0ba2d0666a..83d82a5ffe 100644 --- a/src/armnn/Graph.cpp +++ b/src/armnn/Graph.cpp @@ -67,7 +67,7 @@ Status Graph::Print() const for (auto&& it : TopologicalSort()) { BOOST_LOG_TRIVIAL(info) << it->GetName() << ":" << GetLayerTypeAsCString(it->GetType()) - << ":" << GetComputeDeviceAsCString(it->GetComputeDevice()); + << ":" << it->GetBackendId().Get(); } BOOST_LOG_TRIVIAL(info) << "\n\n"; @@ -260,7 +260,7 @@ void Graph::AddCopyLayers() auto MayNeedCopyLayer = [](const Layer& layer) { // All layers should have been associated with a valid compute device at this point. - BOOST_ASSERT(layer.GetComputeDevice() != Compute::Undefined); + BOOST_ASSERT(layer.GetBackendId() != Compute::Undefined); // Does not need another copy layer if a copy layer is already present. return layer.GetType() != LayerType::MemCopy; }; @@ -276,7 +276,7 @@ void Graph::AddCopyLayers() for (auto&& dstInput : connectionCopy) { Layer& dstLayer = dstInput->GetOwningLayer(); - if (MayNeedCopyLayer(dstLayer) && (dstLayer.GetComputeDevice() != srcLayer->GetComputeDevice())) + if (MayNeedCopyLayer(dstLayer) && (dstLayer.GetBackendId() != srcLayer->GetBackendId())) { // A copy layer is needed in between the source and destination layers. // Record the operation rather than attempting to modify the graph as we go. @@ -288,7 +288,7 @@ void Graph::AddCopyLayers() % dstInput->GetSlotIndex()); MemCopyLayer* const copyLayer = InsertNewLayer<MemCopyLayer>(*dstInput, copyLayerName.c_str()); - copyLayer->SetComputeDevice(dstLayer.GetComputeDevice()); + copyLayer->SetBackendId(dstLayer.GetBackendId()); } } ++srcOutputIndex; |