Skip to content

Commit 95e78b2

Browse files
Schema header added to micro_allocator_test.cc
1 parent fe760f3 commit 95e78b2

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tensorflow/lite/micro/micro_allocator_test.cc

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

3233
namespace tflite {
3334
namespace 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

13901391
TF_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
14541455
TF_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

Comments
 (0)