diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2018-11-15 14:43:10 +0000 |
|---|---|---|
| committer | Anthony Barbier <Anthony.barbier@arm.com> | 2018-11-19 09:58:20 +0000 |
| commit | 32521430ad67172d372721979e8a32b7d4317620 (patch) | |
| tree | c18921290341c1d730fbff406960dc7ea7f61306 /arm_compute/graph/backends/ValidateHelpers.h | |
| parent | d696cb6d18c2fe66f1abce88bbd14faf2137ef89 (diff) | |
| download | ComputeLibrary-32521430ad67172d372721979e8a32b7d4317620.tar.gz | |
COMPMID-1750: Add PriorBox operator to graph API.
Change-Id: I5d2ed5dcc342abff8124762f7bdee587cdf20032
Diffstat (limited to 'arm_compute/graph/backends/ValidateHelpers.h')
| -rw-r--r-- | arm_compute/graph/backends/ValidateHelpers.h | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/arm_compute/graph/backends/ValidateHelpers.h b/arm_compute/graph/backends/ValidateHelpers.h index 7c31a80967..a6864c2286 100644 --- a/arm_compute/graph/backends/ValidateHelpers.h +++ b/arm_compute/graph/backends/ValidateHelpers.h @@ -300,6 +300,29 @@ Status validate_permute_layer(PermuteLayerNode &node) return PermuteLayer::validate(input, output, perm); } +/** Validates a priorbox layer node + * + * @tparam PriorBoxLayer PriorBox layer type + * + * @param[in] node Node to validate + * + * @return Status + */ +template <typename PriorBoxLayer> +Status validate_priorbox_layer(PriorBoxLayerNode &node) +{ + ARM_COMPUTE_LOG_GRAPH_VERBOSE("Validating PriorBoxLayer node with ID : " << node.id() << " and Name: " << node.name() << std::endl); + ARM_COMPUTE_RETURN_ERROR_ON(node.num_inputs() != 2); + ARM_COMPUTE_RETURN_ERROR_ON(node.num_outputs() != 1); + + // Extract IO and info + arm_compute::ITensorInfo *input0 = get_backing_tensor_info(node.input(0)); + arm_compute::ITensorInfo *input1 = get_backing_tensor_info(node.input(1)); + arm_compute::ITensorInfo *output = get_backing_tensor_info(node.output(0)); + const PriorBoxLayerInfo prior_info = node.priorbox_info(); + + return PriorBoxLayer::validate(input0, input1, output, prior_info); +} /** Validates a Reorg layer node * |