diff options
Diffstat (limited to 'src/backends/gpuFsa/GpuFsaLayerSupport.cpp')
-rw-r--r-- | src/backends/gpuFsa/GpuFsaLayerSupport.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp index e98275c166..2e5c7d5a53 100644 --- a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp +++ b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp @@ -13,6 +13,7 @@ #include "layers/GpuFsaConvolution2d.hpp" #include "layers/GpuFsaDepthwiseConvolution2d.hpp" #include "layers/GpuFsaElementwiseBinaryAdd.hpp" +#include "layers/GpuFsaElementwiseBinarySub.hpp" #endif #include <vector> @@ -144,6 +145,13 @@ bool GpuFsaLayerSupport::IsLayerSupported(const LayerType& type, infos[0], infos[1]); } + else if (desc->m_Operation == BinaryOperation::Sub) + { + FORWARD_LAYER_VALIDATE_FUNC(GpuFsaElementwiseBinarySubValidate, + reasonIfUnsupported, + infos[0], + infos[1]); + } else { throw InvalidArgumentException("Invalid ElementwiseBinary BinaryOperation operation."); |