Skip to content

Commit ceaaa3c

Browse files
ddavis-2015veblush
authored andcommitted
Update test_conv_model.cc as the Flatbuffer library was failing to parse the model correctly for cortex-m3-qemu builds.
Update TestAllocatePersistentTfLiteTensor test to match updated test_conv_model.tflite model.
1 parent 189feab commit ceaaa3c

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
@@ -896,14 +896,14 @@ TF_LITE_MICRO_TEST(TestAllocatePersistentTfLiteTensor) {
896896
TF_LITE_MICRO_EXPECT(allocator != nullptr);
897897

898898
TfLiteTensor* tensor1 = allocator->AllocatePersistentTfLiteTensor(
899-
model, /*subgraph_allocations=*/nullptr, /*tensor_index=*/1,
899+
model, /*subgraph_allocations=*/nullptr, /*tensor_index=*/6,
900900
/*subgraph_index=*/0);
901901
TF_LITE_MICRO_EXPECT(tensor1 != nullptr);
902902
TF_LITE_MICRO_EXPECT(tensor1->quantization.params != nullptr);
903903
TF_LITE_MICRO_EXPECT_FALSE(tensor1->is_variable);
904904

905905
TfLiteTensor* tensor2 = allocator->AllocatePersistentTfLiteTensor(
906-
model, /*subgraph_allocations=*/nullptr, /*tensor_index=*/2,
906+
model, /*subgraph_allocations=*/nullptr, /*tensor_index=*/4,
907907
/*subgraph_index=*/0);
908908
TF_LITE_MICRO_EXPECT(tensor2 != nullptr);
909909
TF_LITE_MICRO_EXPECT(tensor2->quantization.params != nullptr);

0 commit comments

Comments
 (0)