| 27 | } |
| 28 | |
| 29 | void Fill3dTest(tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, |
| 30 | float fill = 5.0f ) |
| 31 | { |
| 32 | std::vector<int32_t> inputShape { 3 }; |
| 33 | std::vector<int32_t> tensorShape { 3, 3, 3 }; |
| 34 | std::vector<float> expectedOutputValues = { fill, fill, fill, |
| 35 | fill, fill, fill, |
| 36 | fill, fill, fill, |
| 37 | |
| 38 | fill, fill, fill, |
| 39 | fill, fill, fill, |
| 40 | fill, fill, fill, |
| 41 | |
| 42 | fill, fill, fill, |
| 43 | fill, fill, fill, |
| 44 | fill, fill, fill }; |
| 45 | |
| 46 | FillTest<float>(fillOperatorCode, |
| 47 | ::tflite::TensorType_FLOAT32, |
| 48 | inputShape, |
| 49 | tensorShape, |
| 50 | expectedOutputValues, |
| 51 | fill); |
| 52 | } |
| 53 | |
| 54 | void Fill4dTest(tflite::BuiltinOperator fillOperatorCode = tflite::BuiltinOperator_FILL, |
| 55 | float fill = 3.0f ) |