diff options
| author | Alex Gilday <alexander.gilday@arm.com> | 2018-02-15 11:07:18 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:47:18 +0000 |
| commit | 7bfb199c1461fd553a78ca1947855c7d25106803 (patch) | |
| tree | ac7b3e83cf6866f82cdbe87cfe4fa6c4bdd9e3e9 /src/graph/operations/CLSimpleOperations.cpp | |
| parent | dde9ec96f471127e5b6d8dfaeffce024b6326f1a (diff) | |
| download | ComputeLibrary-7bfb199c1461fd553a78ca1947855c7d25106803.tar.gz | |
COMPMID-915: Create ResNet50 example
ResidualLayer node (COMPMID-916) also created as required for the ResNet
architecture.
Change-Id: I3aef0b6d6fd5bfcd4916fed4d8d4466b8a92b70d
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/120562
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/graph/operations/CLSimpleOperations.cpp')
| -rw-r--r-- | src/graph/operations/CLSimpleOperations.cpp | 30 |
1 files changed, 29 insertions, 1 deletions
diff --git a/src/graph/operations/CLSimpleOperations.cpp b/src/graph/operations/CLSimpleOperations.cpp index 94e3fe15f7..fe56122009 100644 --- a/src/graph/operations/CLSimpleOperations.cpp +++ b/src/graph/operations/CLSimpleOperations.cpp @@ -66,6 +66,34 @@ REGISTER_SIMPLE_OPERATION(CLActivationLayerOperation, OPENCL, OperationType::Act return std::move(activation); } +/* Arithmetic addition */ +REGISTER_SIMPLE_OPERATION(CLArithmeticAdditionOperation, OPENCL, OperationType::ArithmeticAddition) +{ + ARM_COMPUTE_ERROR_ON(ctx.num_inputs() != 2); + ARM_COMPUTE_ERROR_ON(ctx.num_outputs() != 1); + ARM_COMPUTE_ERROR_ON(dynamic_cast<arm_compute::ICLTensor *>(ctx.input(0)) == nullptr); + ARM_COMPUTE_ERROR_ON(dynamic_cast<arm_compute::ICLTensor *>(ctx.input(1)) == nullptr); + ARM_COMPUTE_ERROR_ON(dynamic_cast<arm_compute::ICLTensor *>(ctx.output(0)) == nullptr); + + // Extract IO and info + auto *in1 = dynamic_cast<arm_compute::ICLTensor *>(ctx.input(0)); + auto *in2 = dynamic_cast<arm_compute::ICLTensor *>(ctx.input(1)); + auto *out = dynamic_cast<arm_compute::ICLTensor *>(ctx.output(0)); + + auto addition = arm_compute::support::cpp14::make_unique<arm_compute::CLArithmeticAddition>(); + addition->configure(in1, in2, out, ConvertPolicy::SATURATE); + + // Log info + ARM_COMPUTE_LOG_GRAPH_INFO("Instantiating CLArithmeticAddition" + << " Data Type: " << in1->info()->data_type() + << " Input 1 shape: " << in1->info()->tensor_shape() + << " Input 2 shape: " << in2->info()->tensor_shape() + << " Output shape: " << out->info()->tensor_shape() + << std::endl); + + return std::move(addition); +} + /* Batch Normalization Layer */ REGISTER_SIMPLE_OPERATION(CLBatchNormalizationLayerOperation, OPENCL, OperationType::BatchNormalizationLayer) { @@ -464,4 +492,4 @@ REGISTER_SIMPLE_OPERATION(CLSoftmaxLayerOperation, OPENCL, OperationType::Softma << std::endl); return std::move(smx); -}
\ No newline at end of file +} |