1 /* 2 * Copyright (c) 2017 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_GOOGLENET_GEMM_DATASET 25 #define ARM_COMPUTE_TEST_GOOGLENET_GEMM_DATASET 26 27 #include "tests/datasets/GEMMDataset.h" 28 29 #include "utils/TypePrinter.h" 30 31 #include "arm_compute/core/TensorShape.h" 32 33 namespace arm_compute 34 { 35 namespace test 36 { 37 namespace datasets 38 { 39 class GoogleNetGEMMDataset final : public GEMMDataset 40 { 41 public: GoogleNetGEMMDataset()42 GoogleNetGEMMDataset() 43 { 44 add_config(TensorShape{ 147U, 12544U }, TensorShape{ 64U, 147U }, TensorShape{ 64U, 12544U }, TensorShape{ 64U, 12544U }, 1.f, 0.f); 45 add_config(TensorShape{ 192U, 784U }, TensorShape{ 96U, 192U }, TensorShape{ 96U, 784U }, TensorShape{ 96U, 784U }, 1.f, 0.f); 46 add_config(TensorShape{ 192U, 784U }, TensorShape{ 32U, 192U }, TensorShape{ 32U, 784U }, TensorShape{ 32U, 784U }, 1.f, 0.f); 47 add_config(TensorShape{ 256U, 784U }, TensorShape{ 32U, 256U }, TensorShape{ 32U, 784U }, TensorShape{ 32U, 784U }, 1.f, 0.f); 48 add_config(TensorShape{ 480U, 196U }, TensorShape{ 96U, 480U }, TensorShape{ 96U, 196U }, TensorShape{ 96U, 196U }, 1.f, 0.f); 49 add_config(TensorShape{ 480U, 196U }, TensorShape{ 64U, 480U }, TensorShape{ 64U, 196U }, TensorShape{ 64U, 196U }, 1.f, 0.f); 50 add_config(TensorShape{ 508U, 196U }, TensorShape{ 24U, 508U }, TensorShape{ 24U, 196U }, TensorShape{ 24U, 196U }, 1.f, 0.f); 51 add_config(TensorShape{ 512U, 196U }, TensorShape{ 128U, 512U }, TensorShape{ 128U, 196U }, TensorShape{ 128U, 196U }, 1.f, 0.f); 52 add_config(TensorShape{ 512U, 196U }, TensorShape{ 64U, 512U }, TensorShape{ 64U, 196U }, TensorShape{ 64U, 196U }, 1.f, 0.f); 53 add_config(TensorShape{ 512U, 196U }, TensorShape{ 32U, 512U }, TensorShape{ 32U, 196U }, TensorShape{ 32U, 196U }, 1.f, 0.f); 54 add_config(TensorShape{ 528U, 196U }, TensorShape{ 160U, 528U }, TensorShape{ 160U, 196U }, TensorShape{ 160U, 196U }, 1.f, 0.f); 55 add_config(TensorShape{ 528U, 196U }, TensorShape{ 128U, 528U }, TensorShape{ 128U, 196U }, TensorShape{ 128U, 196U }, 1.f, 0.f); 56 add_config(TensorShape{ 832U, 49U }, TensorShape{ 48U, 832U }, TensorShape{ 48U, 49U }, TensorShape{ 48U, 49U }, 1.f, 0.f); 57 add_config(TensorShape{ 832U, 49U }, TensorShape{ 192U, 832U }, TensorShape{ 192U, 49U }, TensorShape{ 192U, 49U }, 1.f, 0.f); 58 add_config(TensorShape{ 832U, 49U }, TensorShape{ 128U, 832U }, TensorShape{ 128U, 49U }, TensorShape{ 128U, 49U }, 1.f, 0.f); 59 add_config(TensorShape{ 528U, 16U }, TensorShape{ 128U, 528U }, TensorShape{ 128U, 16U }, TensorShape{ 128U, 16U }, 1.f, 0.f); 60 add_config(TensorShape{ 64U, 3136U }, TensorShape{ 64U, 64U }, TensorShape{ 64U, 3136U }, TensorShape{ 64U, 3136U }, 1.f, 0.f); 61 add_config(TensorShape{ 864U, 784U }, TensorShape{ 128U, 864U }, TensorShape{ 128U, 784U }, TensorShape{ 128U, 784U }, 1.f, 0.f); 62 add_config(TensorShape{ 256U, 784U }, TensorShape{ 128U, 256U }, TensorShape{ 128U, 784U }, TensorShape{ 128U, 784U }, 1.f, 0.f); 63 add_config(TensorShape{ 800U, 784U }, TensorShape{ 96U, 800U }, TensorShape{ 96U, 784U }, TensorShape{ 96U, 784U }, 1.f, 0.f); 64 add_config(TensorShape{ 864U, 196U }, TensorShape{ 204U, 864U }, TensorShape{ 204U, 196U }, TensorShape{ 204U, 196U }, 1.f, 0.f); 65 add_config(TensorShape{ 508U, 196U }, TensorShape{ 160U, 508U }, TensorShape{ 160U, 196U }, TensorShape{ 160U, 196U }, 1.f, 0.f); 66 add_config(TensorShape{ 600U, 196U }, TensorShape{ 64U, 600U }, TensorShape{ 64U, 196U }, TensorShape{ 64U, 196U }, 1.f, 0.f); 67 add_config(TensorShape{ 1152U, 196U }, TensorShape{ 256U, 1152U }, TensorShape{ 256U, 196U }, TensorShape{ 256U, 196U }, 1.f, 0.f); 68 add_config(TensorShape{ 512U, 196U }, TensorShape{ 112U, 512U }, TensorShape{ 112U, 196U }, TensorShape{ 112U, 196U }, 1.f, 0.f); 69 add_config(TensorShape{ 800U, 196U }, TensorShape{ 64U, 800U }, TensorShape{ 64U, 196U }, TensorShape{ 64U, 196U }, 1.f, 0.f); 70 add_config(TensorShape{ 1440U, 196U }, TensorShape{ 320U, 1440U }, TensorShape{ 320U, 196U }, TensorShape{ 320U, 196U }, 1.f, 0.f); 71 add_config(TensorShape{ 832U, 49U }, TensorShape{ 256U, 832U }, TensorShape{ 256U, 49U }, TensorShape{ 256U, 49U }, 1.f, 0.f); 72 add_config(TensorShape{ 1200U, 49U }, TensorShape{ 128U, 1200U }, TensorShape{ 128U, 49U }, TensorShape{ 128U, 49U }, 1.f, 0.f); 73 add_config(TensorShape{ 1728U, 49U }, TensorShape{ 384U, 1728U }, TensorShape{ 384U, 49U }, TensorShape{ 384U, 49U }, 1.f, 0.f); 74 add_config(TensorShape{ 508U, 16U }, TensorShape{ 128U, 508U }, TensorShape{ 128U, 16U }, TensorShape{ 128U, 16U }, 1.f, 0.f); 75 add_config(TensorShape{ 2048U, 1U }, TensorShape{ 1024U, 2048U }, TensorShape{ 1024U, 1U }, TensorShape{ 1024U, 1U }, 1.f, 0.f); 76 add_config(TensorShape{ 576U, 3136U }, TensorShape{ 192U, 576U }, TensorShape{ 192U, 3136U }, TensorShape{ 192U, 3136U }, 1.f, 0.f); 77 add_config(TensorShape{ 192U, 784U }, TensorShape{ 16U, 192U }, TensorShape{ 16U, 784U }, TensorShape{ 16U, 784U }, 1.f, 0.f); 78 add_config(TensorShape{ 256U, 784U }, TensorShape{ 128U, 256U }, TensorShape{ 128U, 784U }, TensorShape{ 128U, 784U }, 1.f, 0.f); 79 add_config(TensorShape{ 256U, 784U }, TensorShape{ 64U, 256U }, TensorShape{ 64U, 784U }, TensorShape{ 64U, 784U }, 1.f, 0.f); 80 add_config(TensorShape{ 480U, 196U }, TensorShape{ 16U, 480U }, TensorShape{ 16U, 196U }, TensorShape{ 16U, 196U }, 1.f, 0.f); 81 add_config(TensorShape{ 508U, 196U }, TensorShape{ 112U, 508U }, TensorShape{ 112U, 196U }, TensorShape{ 112U, 196U }, 1.f, 0.f); 82 add_config(TensorShape{ 508U, 196U }, TensorShape{ 64U, 508U }, TensorShape{ 64U, 196U }, TensorShape{ 64U, 196U }, 1.f, 0.f); 83 add_config(TensorShape{ 512U, 196U }, TensorShape{ 24U, 512U }, TensorShape{ 24U, 196U }, TensorShape{ 24U, 196U }, 1.f, 0.f); 84 add_config(TensorShape{ 512U, 196U }, TensorShape{ 144U, 512U }, TensorShape{ 144U, 196U }, TensorShape{ 144U, 196U }, 1.f, 0.f); 85 add_config(TensorShape{ 512U, 196U }, TensorShape{ 64U, 512U }, TensorShape{ 64U, 196U }, TensorShape{ 64U, 196U }, 1.f, 0.f); 86 add_config(TensorShape{ 528U, 196U }, TensorShape{ 32U, 528U }, TensorShape{ 32U, 196U }, TensorShape{ 32U, 196U }, 1.f, 0.f); 87 add_config(TensorShape{ 832U, 49U }, TensorShape{ 160U, 832U }, TensorShape{ 160U, 49U }, TensorShape{ 160U, 49U }, 1.f, 0.f); 88 add_config(TensorShape{ 832U, 49U }, TensorShape{ 128U, 832U }, TensorShape{ 128U, 49U }, TensorShape{ 128U, 49U }, 1.f, 0.f); 89 add_config(TensorShape{ 832U, 49U }, TensorShape{ 48U, 832U }, TensorShape{ 48U, 49U }, TensorShape{ 48U, 49U }, 1.f, 0.f); 90 add_config(TensorShape{ 2048U, 1U }, TensorShape{ 1024U, 2048U }, TensorShape{ 1024U, 1U }, TensorShape{ 1024U, 1U }, 1.f, 0.f); 91 add_config(TensorShape{ 1024U, 1U }, TensorShape{ 1008U, 1024U }, TensorShape{ 1008U, 1U }, TensorShape{ 1008U, 1U }, 1.f, 0.f); 92 } 93 }; 94 } // namespace datasets 95 } // namespace test 96 } // namespace arm_compute 97 #endif /* ARM_COMPUTE_TEST_GOOGLENET_GEMM_DATASET */ 98