xref: /aosp_15_r20/external/ComputeLibrary/tests/IAccessor.h (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1*c217d954SCole Faust /*
2*c217d954SCole Faust  * Copyright (c) 2017-2019 Arm Limited.
3*c217d954SCole Faust  *
4*c217d954SCole Faust  * SPDX-License-Identifier: MIT
5*c217d954SCole Faust  *
6*c217d954SCole Faust  * Permission is hereby granted, free of charge, to any person obtaining a copy
7*c217d954SCole Faust  * of this software and associated documentation files (the "Software"), to
8*c217d954SCole Faust  * deal in the Software without restriction, including without limitation the
9*c217d954SCole Faust  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10*c217d954SCole Faust  * sell copies of the Software, and to permit persons to whom the Software is
11*c217d954SCole Faust  * furnished to do so, subject to the following conditions:
12*c217d954SCole Faust  *
13*c217d954SCole Faust  * The above copyright notice and this permission notice shall be included in all
14*c217d954SCole Faust  * copies or substantial portions of the Software.
15*c217d954SCole Faust  *
16*c217d954SCole Faust  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17*c217d954SCole Faust  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18*c217d954SCole Faust  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19*c217d954SCole Faust  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20*c217d954SCole Faust  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21*c217d954SCole Faust  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22*c217d954SCole Faust  * SOFTWARE.
23*c217d954SCole Faust  */
24*c217d954SCole Faust #ifndef ARM_COMPUTE_TEST_IACCESSOR_H
25*c217d954SCole Faust #define ARM_COMPUTE_TEST_IACCESSOR_H
26*c217d954SCole Faust 
27*c217d954SCole Faust #include "arm_compute/core/Coordinates.h"
28*c217d954SCole Faust #include "arm_compute/core/Types.h"
29*c217d954SCole Faust 
30*c217d954SCole Faust namespace arm_compute
31*c217d954SCole Faust {
32*c217d954SCole Faust namespace test
33*c217d954SCole Faust {
34*c217d954SCole Faust /** Common interface to provide information and access to tensor like
35*c217d954SCole Faust  * structures.
36*c217d954SCole Faust  */
37*c217d954SCole Faust class IAccessor
38*c217d954SCole Faust {
39*c217d954SCole Faust public:
40*c217d954SCole Faust     /** Virtual destructor. */
41*c217d954SCole Faust     virtual ~IAccessor() = default;
42*c217d954SCole Faust 
43*c217d954SCole Faust     /** Shape of the tensor.
44*c217d954SCole Faust      *
45*c217d954SCole Faust      * @return the shape of the tensor.
46*c217d954SCole Faust      */
47*c217d954SCole Faust     virtual TensorShape shape() const = 0;
48*c217d954SCole Faust 
49*c217d954SCole Faust     /** Size of each element in the tensor in bytes.
50*c217d954SCole Faust      *
51*c217d954SCole Faust      * @return the size of each element in the tensor in bytes.
52*c217d954SCole Faust      */
53*c217d954SCole Faust     virtual size_t element_size() const = 0;
54*c217d954SCole Faust 
55*c217d954SCole Faust     /** Total size of the tensor in bytes.
56*c217d954SCole Faust      *
57*c217d954SCole Faust      * @return the total size of the tensor in bytes.
58*c217d954SCole Faust      */
59*c217d954SCole Faust     virtual size_t size() const = 0;
60*c217d954SCole Faust 
61*c217d954SCole Faust     /** Image format of the tensor.
62*c217d954SCole Faust      *
63*c217d954SCole Faust      * @return the format of the tensor.
64*c217d954SCole Faust      */
65*c217d954SCole Faust     virtual Format format() const = 0;
66*c217d954SCole Faust 
67*c217d954SCole Faust     /** Data layout of the tensor.
68*c217d954SCole Faust      *
69*c217d954SCole Faust      * @return the data layout of the tensor.
70*c217d954SCole Faust      */
71*c217d954SCole Faust     virtual DataLayout data_layout() const = 0;
72*c217d954SCole Faust 
73*c217d954SCole Faust     /** Data type of the tensor.
74*c217d954SCole Faust      *
75*c217d954SCole Faust      * @return the data type of the tensor.
76*c217d954SCole Faust      */
77*c217d954SCole Faust     virtual DataType data_type() const = 0;
78*c217d954SCole Faust 
79*c217d954SCole Faust     /** Number of channels of the tensor.
80*c217d954SCole Faust      *
81*c217d954SCole Faust      * @return the number of channels of the tensor.
82*c217d954SCole Faust      */
83*c217d954SCole Faust     virtual int num_channels() const = 0;
84*c217d954SCole Faust 
85*c217d954SCole Faust     /** Number of elements of the tensor.
86*c217d954SCole Faust      *
87*c217d954SCole Faust      * @return the number of elements of the tensor.
88*c217d954SCole Faust      */
89*c217d954SCole Faust     virtual int num_elements() const = 0;
90*c217d954SCole Faust 
91*c217d954SCole Faust     /** Available padding around the tensor.
92*c217d954SCole Faust      *
93*c217d954SCole Faust      * @return the available padding around the tensor.
94*c217d954SCole Faust      */
95*c217d954SCole Faust     virtual PaddingSize padding() const = 0;
96*c217d954SCole Faust 
97*c217d954SCole Faust     /** Quantization info in case of asymmetric quantized type
98*c217d954SCole Faust      *
99*c217d954SCole Faust      * @return
100*c217d954SCole Faust      */
101*c217d954SCole Faust     virtual QuantizationInfo quantization_info() const = 0;
102*c217d954SCole Faust 
103*c217d954SCole Faust     /** Read only access to the specified element.
104*c217d954SCole Faust      *
105*c217d954SCole Faust      * @param[in] coord Coordinates of the desired element.
106*c217d954SCole Faust      *
107*c217d954SCole Faust      * @return A pointer to the desired element.
108*c217d954SCole Faust      */
109*c217d954SCole Faust     virtual const void *operator()(const Coordinates &coord) const = 0;
110*c217d954SCole Faust 
111*c217d954SCole Faust     /** Access to the specified element.
112*c217d954SCole Faust      *
113*c217d954SCole Faust      * @param[in] coord Coordinates of the desired element.
114*c217d954SCole Faust      *
115*c217d954SCole Faust      * @return A pointer to the desired element.
116*c217d954SCole Faust      */
117*c217d954SCole Faust     virtual void *operator()(const Coordinates &coord) = 0;
118*c217d954SCole Faust };
119*c217d954SCole Faust } // namespace test
120*c217d954SCole Faust } // namespace arm_compute
121*c217d954SCole Faust #endif /* ARM_COMPUTE_TEST_IACCESSOR_H */
122