diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-12-24 11:24:40 +0000 |
---|---|---|
committer | Anthony Barbier <Anthony.barbier@arm.com> | 2018-12-27 15:07:40 +0000 |
commit | 985b5d51844ca824b1b7fb816e4098c6bb86e9d4 (patch) | |
tree | 13866e6ebecf1a4bd530545c993a40d8ff6fbf02 /src/runtime/CL/functions/CLUnstack.cpp | |
parent | b5580c0d4d6b7d73c1b75cbc3bb4502cc0e1abe6 (diff) | |
download | ComputeLibrary-985b5d51844ca824b1b7fb816e4098c6bb86e9d4.tar.gz |
COMPMID-1710: Fixed unused function warning in CLUnstack
Change-Id: I94ef19271b059fafb7dad26fee0e229d7e65f64e
Reviewed-on: https://review.mlplatform.org/441
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <Anthony.barbier@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLUnstack.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLUnstack.cpp | 24 |
1 files changed, 11 insertions, 13 deletions
diff --git a/src/runtime/CL/functions/CLUnstack.cpp b/src/runtime/CL/functions/CLUnstack.cpp index 2c8a310043..428d09148b 100644 --- a/src/runtime/CL/functions/CLUnstack.cpp +++ b/src/runtime/CL/functions/CLUnstack.cpp @@ -51,17 +51,6 @@ inline void setup_slice_coordinates_and_mask(Coordinates &slice_start, int32_t & } slice_end_mask = arm_compute::helpers::tensor_transform::construct_slice_end_mask(slice_end); } - -inline const std::vector<ITensorInfo *> get_tensor_info(const std::vector<ICLTensor *> &output_vector) -{ - // given a vector of tensors this function returns a vector to their ITensorInfo* - std::vector<ITensorInfo *> out_info(output_vector.size()); - std::transform(output_vector.begin(), output_vector.end(), out_info.begin(), [](ICLTensor * t) - { - return t->info(); - }); - return out_info; -} } // namespace CLUnstack::CLUnstack() // NOLINT @@ -72,12 +61,21 @@ CLUnstack::CLUnstack() // NOLINT void CLUnstack::configure(const ICLTensor *input, const std::vector<ICLTensor *> &output_vector, int axis) { + std::vector<ITensorInfo *> outputs_vector_info(output_vector.size()); + std::transform(output_vector.begin(), output_vector.end(), outputs_vector_info.begin(), [](ICLTensor * t) + { + ARM_COMPUTE_ERROR_ON_NULLPTR(t); + return t->info(); + }); + ARM_COMPUTE_ERROR_ON_NULLPTR(input); - ARM_COMPUTE_ERROR_THROW_ON(CLUnstack::validate(input->info(), get_tensor_info(output_vector), axis)); + ARM_COMPUTE_ERROR_THROW_ON(CLUnstack::validate(input->info(), outputs_vector_info, axis)); + // Wrap around negative values const unsigned int axis_u = wrap_axis(axis, input->info()); - _num_slices = std::min(output_vector.size(), input->info()->dimension(axis_u)); + _num_slices = std::min(outputs_vector_info.size(), input->info()->dimension(axis_u)); _strided_slice_vector = arm_compute::support::cpp14::make_unique<CLStridedSlice[]>(_num_slices); + Coordinates slice_start; int32_t slice_end_mask; setup_slice_coordinates_and_mask(slice_start, slice_end_mask, input->info()->tensor_shape().num_dimensions()); |