| 55 | } |
| 56 | |
| 57 | void Pad3dTest(tflite::BuiltinOperator padOperatorCode = tflite::BuiltinOperator_PAD, |
| 58 | float pad = 0.0f) |
| 59 | { |
| 60 | // Set input data |
| 61 | std::vector<int32_t> inputShape { 2, 2, 2 }; |
| 62 | std::vector<int32_t> outputShape { 3, 5, 6 }; |
| 63 | std::vector<int32_t> paddingShape { 3, 2 }; |
| 64 | |
| 65 | std::vector<float> inputValues = { 0.0f, 4.0f, |
| 66 | 2.0f, 5.0f, |
| 67 | 6.0f, 1.0f, |
| 68 | 5.0f, 2.0f }; |
| 69 | |
| 70 | std::vector<float> expectedOutputValues = { pad, pad, pad, pad, pad, pad, |
| 71 | pad, pad, pad, pad, pad, pad, |
| 72 | pad, pad, 0.0f, 4.0f, pad, pad, |
| 73 | pad, pad, 2.0f, 5.0f, pad, pad, |
| 74 | pad, pad, pad, pad, pad, pad, |
| 75 | |
| 76 | pad, pad, pad, pad, pad, pad, |
| 77 | pad, pad, pad, pad, pad, pad, |
| 78 | pad, pad, 6.0f, 1.0f, pad, pad, |
| 79 | pad, pad, 5.0f, 2.0f, pad, pad, |
| 80 | pad, pad, pad, pad, pad, pad, |
| 81 | |
| 82 | pad, pad, pad, pad, pad, pad, |
| 83 | pad, pad, pad, pad, pad, pad, |
| 84 | pad, pad, pad, pad, pad, pad, |
| 85 | pad, pad, pad, pad, pad, pad, |
| 86 | pad, pad, pad, pad, pad, pad }; |
| 87 | |
| 88 | std::vector<int32_t> paddingDim = { 0, 1, 2, 1, 2, 2 }; |
| 89 | |
| 90 | PadTest<float>(padOperatorCode, |
| 91 | ::tflite::TensorType_FLOAT32, |
| 92 | inputShape, |
| 93 | paddingShape, |
| 94 | outputShape, |
| 95 | inputValues, |
| 96 | paddingDim, |
| 97 | expectedOutputValues, |
| 98 | pad); |
| 99 | } |
| 100 | |
| 101 | void Pad4dTest(tflite::BuiltinOperator padOperatorCode = tflite::BuiltinOperator_PAD, |
| 102 | float pad = 0.0f) |