tests/GLES_COMPUTE/Helper.h

   1 /*
   2  * Copyright (c) 2017-2018 ARM Limited.
   3  *
   4  * SPDX-License-Identifier: MIT
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a copy
   7  * of this software and associated documentation files (the "Software"), to
   8  * deal in the Software without restriction, including without limitation the
   9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
  10  * sell copies of the Software, and to permit persons to whom the Software is
  11  * furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in all
  14  * copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  22  * SOFTWARE.
  23  */
  24 #ifndef __ARM_COMPUTE_TEST_GC_HELPER_H__
  25 #define __ARM_COMPUTE_TEST_GC_HELPER_H__
  26
  27 #include "tests/Globals.h"
  28
  29 #include "arm_compute/core/ITensor.h"
  30 #include "arm_compute/runtime/GLES_COMPUTE/GCTensor.h"
  31
  32 #include <iostream>
  33
  34 namespace arm_compute
  35 {
  36 namespace test
  37 {
  38 /** Helper to create an empty tensor.
  39  *
  40  * @param[in] shape                Desired shape.
  41  * @param[in] data_type            Desired data type.
  42  * @param[in] num_channels         (Optional) It indicates the number of channels for each tensor element
  43  * @param[in] fixed_point_position (Optional) Fixed point position that expresses the number of bits for the fractional part of the number when the tensor's data type is QS8 or QS16.
  44  *
  45  * @return Empty @ref GCTensor with the specified shape and data type.
  46  */
  47 inline GCTensor create_tensor(const TensorShape &shape, DataType data_type, int num_channels = 1, int fixed_point_position = 0)
  48 {
  49     GCTensor tensor;
  50     tensor.allocator()->init(TensorInfo(shape, num_channels, data_type, fixed_point_position));
  51
  52     return tensor;
  53 }
  54
  55 /** Helper to create an empty tensor.
  56  *
  57  * @param[in] name      File name from which to get the dimensions.
  58  * @param[in] data_type Desired data type.
  59  *
  60  * @return Empty @ref GCTensor with the specified shape and data type.
  61  */
  62 inline GCTensor create_tensor(const std::string &name, DataType data_type)
  63 {
  64     constexpr unsigned int num_channels = 1;
  65
  66     const RawTensor &raw = library->get(name);
  67
  68     GCTensor tensor;
  69     tensor.allocator()->init(TensorInfo(raw.shape(), num_channels, data_type));
  70
  71     return tensor;
  72 }
  73
  74 /** Helper to print tensor.
  75  *
  76  * @param[in] tensor Tensor to print.
  77  * @param[in] name   Tensor name.
  78  * @param[in] info   Format information.
  79  *
  80  * @return Empty @ref GCTensor with the specified shape and data type.
  81  */
  82 inline void print_tensor(ITensor &tensor, const std::string &name, IOFormatInfo info = IOFormatInfo(IOFormatInfo::PrintRegion::Full))
  83 {
  84     std::ostringstream s;
  85     IGCTensor         &t = dynamic_cast<IGCTensor &>(tensor);
  86     t.map();
  87     t.print(s, info);
  88
  89     std::cout << name << ":" << std::endl;
  90     std::cout << s.str().c_str();
  91     t.unmap();
  92 }
  93 } // namespace test
  94 } // namespace arm_compute
  95 #endif /* __ARM_COMPUTE_TEST_GC_HELPER_H__ */