@@ -28,6 +28,7 @@ limitations under the License.
2828#include " tensorflow/lite/micro/test_helpers.h"
2929#include " tensorflow/lite/micro/testing/micro_test.h"
3030#include " tensorflow/lite/micro/testing/test_conv_model.h"
31+ #include " tensorflow/lite/schema/schema_generated.h"
3132
3233namespace tflite {
3334namespace testing {
@@ -1385,7 +1386,7 @@ TF_LITE_MICRO_TEST(TestMultiSubgraphNumScratchAllocations) {
13851386}
13861387
13871388// New tests validating invalid buffer index guards in tensor initialization
1388- // and model allocation paths. These exercise issues #3077, #3076, #3075.
1389+ // and model allocation paths.
13891390
13901391TF_LITE_MICRO_TEST (InitializeTfLiteTensorFromFlatbuffer_InvalidBufferIndexReturnsError) {
13911392 // Arena and allocator for temporary/persistent allocations used by init.
@@ -1454,8 +1455,8 @@ TF_LITE_MICRO_TEST(InitializeTfLiteEvalTensorFromFlatbuffer_InvalidBufferIndexRe
14541455TF_LITE_MICRO_TEST (StartModelAllocation_FailsWhenSubgraphHasTensorWithInvalidBufferIndex) {
14551456 // Build a minimal model with a single tensor that references an invalid
14561457 // buffer index. Make it an output tensor to mimic output-table invalid case
1457- // (#3075), but the allocator will validate all tensors equally, so this also
1458- // simulates intermediate cases (#3076) .
1458+ // but the allocator will validate all tensors equally, so this also
1459+ // simulates intermediate cases.
14591460 flatbuffers::FlatBufferBuilder fbb;
14601461
14611462 // One empty buffer at index 0 in Model.buffers().
0 commit comments