24.02.1
|
Go to the documentation of this file.
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));
TensorShape shape() const override
Shape of the tensor.
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.
__global uchar * offset(const Image *img, int x, int y)
Get the pointer position of a Image.
T x() const
Alias to access the size of the first dimension.
int num_elements() const override
Number of elements of the tensor.
Simple tensor object that stores elements in a consecutive chunk of memory.
SimpleTensor< T > select(const SimpleTensor< uint8_t > &c, const SimpleTensor< T > &x, const SimpleTensor< T > &y)
Copyright (c) 2017-2024 Arm Limited.
const T * data() const
Constant pointer to the underlying buffer.
unsigned int num_dimensions() const
Returns the effective dimensionality of the tensor.