tests/nnfw_api/src/one_op_tests/PadV2.cc

   1 /*
   2  * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include "GenModelTest.h"
  18
  19 TEST_F(GenModelTest, OneOp_PadV2)
  20 {
  21   CircleGen cgen;
  22   int in = cgen.addTensor({{1, 2, 2, 1}, circle::TensorType::TensorType_FLOAT32});
  23   std::vector<int32_t> padding_data{0, 0, 1, 1, 1, 1, 0, 0};
  24   uint32_t padding_buf = cgen.addBuffer(padding_data);
  25   int padding = cgen.addTensor({{4, 2}, circle::TensorType::TensorType_INT32, padding_buf});
  26   std::vector<float> padding_value_data{3.0};
  27   uint32_t padding_value_buf = cgen.addBuffer(padding_value_data);
  28   int padding_value =
  29       cgen.addTensor({{1}, circle::TensorType::TensorType_FLOAT32, padding_value_buf});
  30
  31   int out = cgen.addTensor({{1, 4, 4, 1}, circle::TensorType::TensorType_FLOAT32});
  32
  33   cgen.addOperatorPadV2({{in, padding, padding_value}, {out}});
  34   cgen.setInputsAndOutputs({in}, {out});
  35
  36   _context = std::make_unique<GenModelTestContext>(cgen.finish());
  37   _context->addTestCase(
  38       uniformTCD<float>({{1, 2, 3, 4}}, {{3, 3, 3, 3, 3, 1, 2, 3, 3, 3, 4, 3, 3, 3, 3, 3}}));
  39   _context->setBackends({"cpu"});
  40
  41   SUCCEED();
  42 }
  43
  44 TEST_F(GenModelTest, neg_OneOp_PadV2_InvalidPadRank)
  45 {
  46   CircleGen cgen;
  47   int in = cgen.addTensor({{1, 2, 2, 1}, circle::TensorType::TensorType_FLOAT32});
  48   std::vector<int32_t> padding_data{1, 1, 1, 1};
  49   uint32_t padding_buf = cgen.addBuffer(padding_data);
  50   int padding = cgen.addTensor({{4}, circle::TensorType::TensorType_INT32, padding_buf});
  51   std::vector<float> padding_value_data{3.0};
  52   uint32_t padding_value_buf = cgen.addBuffer(padding_value_data);
  53   int padding_value =
  54       cgen.addTensor({{1}, circle::TensorType::TensorType_FLOAT32, padding_value_buf});
  55
  56   int out = cgen.addTensor({{1, 4, 4, 1}, circle::TensorType::TensorType_FLOAT32});
  57
  58   cgen.addOperatorPad({{in, padding, padding_value}, {out}});
  59   cgen.setInputsAndOutputs({in}, {out});
  60
  61   _context = std::make_unique<GenModelTestContext>(cgen.finish());
  62   _context->setBackends({"acl_cl", "acl_neon", "cpu"});
  63   _context->expectFailCompile();
  64
  65   SUCCEED();
  66 }
  67
  68 TEST_F(GenModelTest, neg_OneOp_PadV2_InvalidPadDim0)
  69 {
  70   CircleGen cgen;
  71   int in = cgen.addTensor({{1, 2, 2, 1}, circle::TensorType::TensorType_FLOAT32});
  72   std::vector<int32_t> padding_data{1, 1, 1, 1};
  73   uint32_t padding_buf = cgen.addBuffer(padding_data);
  74   int padding = cgen.addTensor({{2, 2}, circle::TensorType::TensorType_INT32, padding_buf});
  75   std::vector<float> padding_value_data{3.0};
  76   uint32_t padding_value_buf = cgen.addBuffer(padding_value_data);
  77   int padding_value =
  78       cgen.addTensor({{1}, circle::TensorType::TensorType_FLOAT32, padding_value_buf});
  79
  80   int out = cgen.addTensor({{1, 4, 4, 1}, circle::TensorType::TensorType_FLOAT32});
  81
  82   cgen.addOperatorPad({{in, padding, padding_value}, {out}});
  83   cgen.setInputsAndOutputs({in}, {out});
  84
  85   _context = std::make_unique<GenModelTestContext>(cgen.finish());
  86   _context->setBackends({"acl_cl", "acl_neon", "cpu"});
  87   _context->expectFailCompile();
  88
  89   SUCCEED();
  90 }
  91
  92 TEST_F(GenModelTest, neg_OneOp_PadV2_InvalidPadDim1)
  93 {
  94   CircleGen cgen;
  95   int in = cgen.addTensor({{1, 1, 1, 1}, circle::TensorType::TensorType_FLOAT32});
  96   std::vector<int32_t> padding_data{1, 1, 1, 1};
  97   uint32_t padding_buf = cgen.addBuffer(padding_data);
  98   int padding = cgen.addTensor({{4, 1}, circle::TensorType::TensorType_INT32, padding_buf});
  99   std::vector<float> padding_value_data{3.0};
 100   uint32_t padding_value_buf = cgen.addBuffer(padding_value_data);
 101   int padding_value =
 102       cgen.addTensor({{1}, circle::TensorType::TensorType_FLOAT32, padding_value_buf});
 103
 104   int out = cgen.addTensor({{2, 2, 2, 2}, circle::TensorType::TensorType_FLOAT32});
 105
 106   cgen.addOperatorPad({{in, padding, padding_value}, {out}});
 107   cgen.setInputsAndOutputs({in}, {out});
 108
 109   _context = std::make_unique<GenModelTestContext>(cgen.finish());
 110   _context->setBackends({"acl_cl", "acl_neon", "cpu"});
 111   _context->expectFailCompile();
 112
 113   SUCCEED();
 114 }