diff options
Diffstat (limited to 'src/backends/gpuFsa/GpuFsaBackend.cpp')
| -rw-r--r-- | src/backends/gpuFsa/GpuFsaBackend.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/GpuFsaBackend.cpp b/src/backends/gpuFsa/GpuFsaBackend.cpp index 7951b17d9d..8b62aec9e6 100644 --- a/src/backends/gpuFsa/GpuFsaBackend.cpp +++ b/src/backends/gpuFsa/GpuFsaBackend.cpp @@ -23,6 +23,7 @@ #include "layers/GpuFsaConvolution2d.hpp" #include "layers/GpuFsaDepthwiseConvolution2d.hpp" #include "layers/GpuFsaElementwiseBinaryAdd.hpp" +#include "layers/GpuFsaElementwiseBinarySub.hpp" namespace armnn { @@ -305,6 +306,13 @@ OptimizationViews GpuFsaBackend::OptimizeSubgraphView(const SubgraphView& subgra GpuFsaElementwiseBinaryAddCreateOp(preCompiledBlobPtr, input0, input1); } + else if (desc->m_Operation == BinaryOperation::Sub) + { + auto input0 = base.GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(); + auto input1 = base.GetInputSlot(1).GetConnectedOutputSlot()->GetTensorInfo(); + + GpuFsaElementwiseBinarySubCreateOp(preCompiledBlobPtr, input0, input1); + } break; } default: |