| 11 | { |
| 12 | |
| 13 | void Pad2dTest(tflite::BuiltinOperator padOperatorCode = tflite::BuiltinOperator_PAD, |
| 14 | float pad = 0.0f) |
| 15 | { |
| 16 | // Set input data |
| 17 | std::vector<int32_t> inputShape { 2, 2, 2 }; |
| 18 | std::vector<int32_t> outputShape { 3, 5, 6 }; |
| 19 | std::vector<int32_t> paddingShape { 3, 2 }; |
| 20 | |
| 21 | std::vector<float> inputValues = { 0.0f, 4.0f, |
| 22 | 2.0f, -5.0f, |
| 23 | 6.0f, 1.0f, |
| 24 | 5.0f, -2.0f }; |
| 25 | |
| 26 | std::vector<float> expectedOutputValues = { pad, pad, pad, pad, pad, pad, |
| 27 | pad, pad, pad, pad, pad, pad, |
| 28 | pad, pad, 0.0f, 4.0f, pad, pad, |
| 29 | pad, pad, 2.0f, -5.0f, pad, pad, |
| 30 | pad, pad, pad, pad, pad, pad, |
| 31 | |
| 32 | pad, pad, pad, pad, pad, pad, |
| 33 | pad, pad, pad, pad, pad, pad, |
| 34 | pad, pad, 6.0f, 1.0f, pad, pad, |
| 35 | pad, pad, 5.0f, -2.0f, pad, pad, |
| 36 | pad, pad, pad, pad, pad, pad, |
| 37 | |
| 38 | pad, pad, pad, pad, pad, pad, |
| 39 | pad, pad, pad, pad, pad, pad, |
| 40 | pad, pad, pad, pad, pad, pad, |
| 41 | pad, pad, pad, pad, pad, pad, |
| 42 | pad, pad, pad, pad, pad, pad }; |
| 43 | |
| 44 | std::vector<int32_t> paddingDim = { 0, 1, 2, 1, 2, 2 }; |
| 45 | |
| 46 | PadTest<float>(padOperatorCode, |
| 47 | ::tflite::TensorType_FLOAT32, |
| 48 | inputShape, |
| 49 | paddingShape, |
| 50 | outputShape, |
| 51 | inputValues, |
| 52 | paddingDim, |
| 53 | expectedOutputValues, |
| 54 | pad); |
| 55 | } |
| 56 | |
| 57 | void Pad3dTest(tflite::BuiltinOperator padOperatorCode = tflite::BuiltinOperator_PAD, |
| 58 | float pad = 0.0f) |