24.04
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CPU_KERNELS_MAXUNPOOL_GENERIC_NEON_IMPL_H
25 #define ACL_SRC_CPU_KERNELS_MAXUNPOOL_GENERIC_NEON_IMPL_H
38 Iterator indices_itr(indices, window);
39 auto out_ptr =
reinterpret_cast<T *
>(output->
buffer());
45 auto vindices =
reinterpret_cast<uint32_t *
>(indices_itr.
ptr());
46 auto vinput =
reinterpret_cast<T *
>(input_itr.
ptr());
47 out_ptr[
id[3] * out_stride_w /
sizeof(T) + *vindices] = *vinput;
49 input_itr, indices_itr);
53 #endif // ACL_SRC_CPU_KERNELS_MAXUNPOOL_GENERIC_NEON_IMPL_H
Interface for CPU tensor.
Includes all wrapper headers at once.
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
constexpr uint8_t * ptr() const
Return a pointer to the current pixel.
void max_unpooling(const ITensor *input, const ITensor *indices, ITensor *output, const Window &window)
void execute_window_loop(const Window &w, L &&lambda_function, Ts &&...iterators)
Iterate through the passed window, automatically adjusting the iterators and calling the lambda_funct...
Iterator updated by execute_window_loop for each window element.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
virtual const Strides & strides_in_bytes() const =0
The strides in bytes for accessing each dimension of the tensor.
virtual uint8_t * buffer() const =0
Interface to be implemented by the child class to return a pointer to CPU memory.