diff options
Diffstat (limited to 'src/backends/neon')
| -rw-r--r-- | src/backends/neon/NeonBackend.cpp | 4 | ||||
| -rw-r--r-- | src/backends/neon/NeonTensorHandle.hpp | 1 |
2 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp index 7089f23efa..7a258c38e0 100644 --- a/src/backends/neon/NeonBackend.cpp +++ b/src/backends/neon/NeonBackend.cpp @@ -225,14 +225,14 @@ OptimizationViews NeonBackend::OptimizeSubgraphView(const SubgraphView& subgraph if (baseLayer->GetParameters().m_BiasEnabled) { - biases = baseLayer->m_Bias->GetTensorInfo(); + biases = baseLayer->GetInputSlot(2).GetConnectedOutputSlot()->GetTensorInfo(); } arm_compute::Status status = NeonDepthwiseConvolutionWorkloadValidate( baseLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), baseLayer->GetParameters(), - baseLayer->m_Weight->GetTensorInfo(), + baseLayer->GetInputSlot(1).GetConnectedOutputSlot()->GetTensorInfo(), biases, &activationDesc); diff --git a/src/backends/neon/NeonTensorHandle.hpp b/src/backends/neon/NeonTensorHandle.hpp index 9445cb1c75..fcae77cdaa 100644 --- a/src/backends/neon/NeonTensorHandle.hpp +++ b/src/backends/neon/NeonTensorHandle.hpp @@ -241,6 +241,7 @@ private: break; case arm_compute::DataType::QSYMM8: case arm_compute::DataType::QASYMM8_SIGNED: + case arm_compute::DataType::QSYMM8_PER_CHANNEL: armcomputetensorutils::CopyArmComputeITensorData(static_cast<const int8_t*>(memory), this->GetTensor()); break; |