aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/graph')
-rw-r--r--arm_compute/graph/TypePrinter.h5
-rw-r--r--arm_compute/graph/Types.h1
-rw-r--r--arm_compute/graph/backends/FunctionHelpers.h8
-rw-r--r--arm_compute/graph/backends/ValidateHelpers.h6
4 files changed, 17 insertions, 3 deletions
diff --git a/arm_compute/graph/TypePrinter.h b/arm_compute/graph/TypePrinter.h
index 62bacae89f..1037c80157 100644
--- a/arm_compute/graph/TypePrinter.h
+++ b/arm_compute/graph/TypePrinter.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2018-2020 Arm Limited.
+ * Copyright (c) 2018-2021 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -218,6 +218,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const EltwiseOperation &el
case EltwiseOperation::Sub:
os << "Sub";
break;
+ case EltwiseOperation::Div:
+ os << "Div";
+ break;
default:
ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
}
diff --git a/arm_compute/graph/Types.h b/arm_compute/graph/Types.h
index b891c1772f..d1c71f815b 100644
--- a/arm_compute/graph/Types.h
+++ b/arm_compute/graph/Types.h
@@ -106,6 +106,7 @@ enum class EltwiseOperation
Sub, /**< Arithmetic subtraction */
Mul, /**< Arithmetic multiplication */
Max, /**< Arithmetic maximum */
+ Div, /**< Arithmetic division */
};
/** Supported Unary Element-wise operations */
diff --git a/arm_compute/graph/backends/FunctionHelpers.h b/arm_compute/graph/backends/FunctionHelpers.h
index ee5dc3e285..9830290d0f 100644
--- a/arm_compute/graph/backends/FunctionHelpers.h
+++ b/arm_compute/graph/backends/FunctionHelpers.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2018-2020 Arm Limited.
+ * Copyright (c) 2018-2021 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -873,6 +873,12 @@ std::unique_ptr<IFunction> create_eltwise_layer(EltwiseLayerNode &node)
std::string("ElementwiseMaximum"),
input1, input2, output, act_info);
}
+ else if(eltwise_op == EltwiseOperation::Div)
+ {
+ std::tie(func, func_name) = create_named_function<typename EltwiseFunctions::Division>(
+ std::string("ArithmeticDivision"),
+ input1, input2, output, act_info);
+ }
else
{
ARM_COMPUTE_ERROR("Unsupported element-wise operation!");
diff --git a/arm_compute/graph/backends/ValidateHelpers.h b/arm_compute/graph/backends/ValidateHelpers.h
index f8cb1c12e9..93d547b036 100644
--- a/arm_compute/graph/backends/ValidateHelpers.h
+++ b/arm_compute/graph/backends/ValidateHelpers.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2018-2020 Arm Limited.
+ * Copyright (c) 2018-2021 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -694,6 +694,10 @@ Status validate_eltwise_Layer(EltwiseLayerNode &node)
{
return EltwiseLayerFunctions::ElementwiseMax::validate(input1, input2, output, act_info);
}
+ else if(eltwise_op == EltwiseOperation::Div)
+ {
+ return EltwiseLayerFunctions::ArithmeticDivision::validate(input1, input2, output, act_info);
+ }
else
{
ARM_COMPUTE_ERROR("Unsupported element-wise operation!");