56 dst[i] = c[i] > 0 ? x[i] : y[i];
61 T *output_ptr =
dst.data();
67 for(
int i = 0; i < outer_size; ++i)
69 const T *input_ptr = c[i] > 0 ? x.
data() : y.
data();
70 memcpy(output_ptr + offset, input_ptr + offset, inner_size *
sizeof(T));
__global uchar * offset(const Image *img, int x, int y)
Get the pointer position of a Image.
DataType data_type() const override
Data type of the tensor.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
TensorShape shape() const override
Shape of the tensor.
Copyright (c) 2017-2021 Arm Limited.
T x() const
Alias to access the size of the first dimension.
SimpleTensor< T > select(const SimpleTensor< uint8_t > &c, const SimpleTensor< T > &x, const SimpleTensor< T > &y)
Simple tensor object that stores elements in a consecutive chunk of memory.
unsigned int num_dimensions() const
Returns the effective dimensionality of the tensor.
int num_elements() const override
Number of elements of the tensor.
const T * data() const
Constant pointer to the underlying buffer.