24.02.1
|
Tensor accessors to make it easier to interface with arm_gemm. More...
#include <WindowIterator.h>
Public Member Functions | |
TensorAccessor (const ITensor &tensor) | |
Constructor: More... | |
size_t | stride (size_t dim) const |
Get the stride of the dimension dim expressed in number of Ts. More... | |
void | set_stride (size_t dim, size_t size) |
Manually set the stride of a dimension. More... | |
void | set_strides (const Strides &strides) |
Manually set the strides. More... | |
T * | get_ptr (unsigned int x, unsigned int y=0, unsigned int z=0, unsigned int w=0) |
Returns a pointer to the element at coordinates (x,y,z,w) More... | |
T * | operator() (unsigned int x, unsigned int y=0, unsigned int z=0, unsigned int w=0) |
Returns a pointer to the element at coordinates (x,y,z,w) More... | |
T * | first_element () |
Returns a pointer to the first element of the tensor. More... | |
T * | operator() () |
Returns a pointer to the first element of the tensor. More... | |
Tensor accessors to make it easier to interface with arm_gemm.
Definition at line 52 of file WindowIterator.h.
|
inline |
Constructor:
[in] | tensor | Source tensor, must be allocated. |
Definition at line 59 of file WindowIterator.h.
|
inline |
Returns a pointer to the first element of the tensor.
Definition at line 122 of file WindowIterator.h.
Referenced by TensorAccessor< T >::operator()().
|
inline |
Returns a pointer to the element at coordinates (x,y,z,w)
[in] | x | X coordinates |
[in] | y | (optional) Y coordinates |
[in] | z | (optional) Z coordinates |
[in] | w | (optional) W coordinates |
Definition at line 101 of file WindowIterator.h.
References arm_compute::test::validation::w.
Referenced by TensorAccessor< T >::operator()().
|
inline |
Returns a pointer to the first element of the tensor.
Definition at line 131 of file WindowIterator.h.
References TensorAccessor< T >::first_element().
|
inline |
Returns a pointer to the element at coordinates (x,y,z,w)
[in] | x | X coordinates |
[in] | y | (optional) Y coordinates |
[in] | z | (optional) Z coordinates |
[in] | w | (optional) W coordinates |
Definition at line 113 of file WindowIterator.h.
References TensorAccessor< T >::get_ptr(), and arm_compute::test::validation::w.
|
inline |
Manually set the stride of a dimension.
[in] | dim | Dimension of the stride to set. |
[in] | size | Value to set the stride to (in bytes). |
Definition at line 80 of file WindowIterator.h.
|
inline |
Manually set the strides.
[in] | strides | Strides to set |
Definition at line 89 of file WindowIterator.h.
|
inline |
Get the stride of the dimension dim expressed in number of Ts.
[in] | dim | Dimension of the wanted stride. |
Definition at line 69 of file WindowIterator.h.
References ARM_COMPUTE_ERROR_ON.