Skip to content

Commit ffa9021

Browse files
committed
Fix test tensor indexes
1 parent f1ef68b commit ffa9021

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
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)