Skip to content

Commit f8a8b80

Browse files
committed
Update test_conv_model.tflite as the Flatbuffer library was failing to parse it correctly for cortex-m3-qemu builds.
Update TestAllocatePersistentTfLiteTensor test to match updated test_conv_model.tflite model.
1 parent 7843552 commit f8a8b80

File tree

2 files changed

+1724
-1801
lines changed

2 files changed

+1724
-1801
lines changed

tensorflow/lite/micro/micro_allocator_test.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -948,14 +948,14 @@ TF_LITE_MICRO_TEST(TestAllocatePersistentTfLiteTensor) {
948948
TF_LITE_MICRO_EXPECT(allocator != nullptr);
949949

950950
TfLiteTensor* tensor1 = allocator->AllocatePersistentTfLiteTensor(
951-
model, /*subgraph_allocations=*/nullptr, /*tensor_index=*/1,
951+
model, /*subgraph_allocations=*/nullptr, /*tensor_index=*/6,
952952
/*subgraph_index=*/0);
953953
TF_LITE_MICRO_EXPECT(tensor1 != nullptr);
954954
TF_LITE_MICRO_EXPECT(tensor1->quantization.params != nullptr);
955955
TF_LITE_MICRO_EXPECT_FALSE(tensor1->is_variable);
956956

957957
TfLiteTensor* tensor2 = allocator->AllocatePersistentTfLiteTensor(
958-
model, /*subgraph_allocations=*/nullptr, /*tensor_index=*/2,
958+
model, /*subgraph_allocations=*/nullptr, /*tensor_index=*/4,
959959
/*subgraph_index=*/0);
960960
TF_LITE_MICRO_EXPECT(tensor2 != nullptr);
961961
TF_LITE_MICRO_EXPECT(tensor2->quantization.params != nullptr);

0 commit comments

Comments
 (0)