| 85 | } |
| 86 | |
| 87 | void FillInt32Test(tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, |
| 88 | int32_t fill = 2 ) |
| 89 | { |
| 90 | std::vector<int32_t> inputShape { 2 }; |
| 91 | std::vector<int32_t> tensorShape { 2, 2 }; |
| 92 | std::vector<int32_t> expectedOutputValues = { fill, fill, |
| 93 | fill, fill }; |
| 94 | |
| 95 | FillTest<int32_t>(fillOperatorCode, |
| 96 | ::tflite::TensorType_INT32, |
| 97 | inputShape, |
| 98 | tensorShape, |
| 99 | expectedOutputValues, |
| 100 | fill); |
| 101 | } |
| 102 | |
| 103 | TEST_SUITE("FillTests") |
| 104 | { |