diff options
Diffstat (limited to 'src/runtime/NEON/functions/NESplit.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NESplit.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NESplit.cpp b/src/runtime/NEON/functions/NESplit.cpp index e947657934..0373ab6f88 100644 --- a/src/runtime/NEON/functions/NESplit.cpp +++ b/src/runtime/NEON/functions/NESplit.cpp @@ -42,8 +42,8 @@ NESplit::NESplit() void NESplit::configure(const ITensor *input, const std::vector<ITensor *> &outputs, unsigned int axis) { // Create Slice functions - _num_outputs = outputs.size(); - _slice_functions = arm_compute::support::cpp14::make_unique<NESlice[]>(_num_outputs); + _num_outputs = outputs.size(); + _slice_functions.resize(_num_outputs); // Get output shape const TensorShape output_shape = arm_compute::misc::shape_calculator::compute_split_shape(input->info(), axis, _num_outputs); |