| 99 | } |
| 100 | |
| 101 | void Pad4dTest(tflite::BuiltinOperator padOperatorCode = tflite::BuiltinOperator_PAD, |
| 102 | float pad = 0.0f) |
| 103 | { |
| 104 | // Set input data |
| 105 | std::vector<int32_t> inputShape { 2, 2, 3, 2 }; |
| 106 | std::vector<int32_t> outputShape { 4, 5, 7, 4 }; |
| 107 | std::vector<int32_t> paddingShape { 4, 2 }; |
| 108 | |
| 109 | std::vector<float> inputValues = { 0.0f, 1.0f, |
| 110 | 2.0f, 3.0f, |
| 111 | 4.0f, 5.0f, |
| 112 | |
| 113 | 6.0f, 7.0f, |
| 114 | 8.0f, 9.0f, |
| 115 | 10.0f, 11.0f, |
| 116 | |
| 117 | 12.0f, 13.0f, |
| 118 | 14.0f, 15.0f, |
| 119 | 16.0f, 17.0f, |
| 120 | |
| 121 | 18.0f, 19.0f, |
| 122 | 20.0f, 21.0f, |
| 123 | 22.0f, 23.0f }; |
| 124 | |
| 125 | std::vector<float> expectedOutputValues = { pad, pad, pad, pad, |
| 126 | pad, pad, pad, pad, |
| 127 | pad, pad, pad, pad, |
| 128 | pad, pad, pad, pad, |
| 129 | pad, pad, pad, pad, |
| 130 | pad, pad, pad, pad, |
| 131 | pad, pad, pad, pad, |
| 132 | |
| 133 | pad, pad, pad, pad, |
| 134 | pad, pad, pad, pad, |
| 135 | pad, pad, pad, pad, |
| 136 | pad, pad, pad, pad, |
| 137 | pad, pad, pad, pad, |
| 138 | pad, pad, pad, pad, |
| 139 | pad, pad, pad, pad, |
| 140 | |
| 141 | pad, pad, pad, pad, |
| 142 | pad, pad, pad, pad, |
| 143 | pad, pad, pad, pad, |
| 144 | pad, pad, pad, pad, |
| 145 | pad, pad, pad, pad, |
| 146 | pad, pad, pad, pad, |
| 147 | pad, pad, pad, pad, |
| 148 | |
| 149 | pad, pad, pad, pad, |
| 150 | pad, pad, pad, pad, |
| 151 | pad, pad, pad, pad, |
| 152 | pad, pad, pad, pad, |
| 153 | pad, pad, pad, pad, |
| 154 | pad, pad, pad, pad, |
| 155 | pad, pad, pad, pad, |
| 156 | |
| 157 | pad, pad, pad, pad, |
| 158 | pad, pad, pad, pad, |