diff --git a/tensorflow/lite/model_test.cc b/tensorflow/lite/model_test.cc index b9efdf676a8..9a2c68e7c91 100644 --- a/tensorflow/lite/model_test.cc +++ b/tensorflow/lite/model_test.cc @@ -386,7 +386,7 @@ TEST(BasicFlatBufferModel, TestParseModelWithSparseTensor) { ASSERT_EQ(InterpreterBuilder(*model, TrivialResolver())(&interpreter), kTfLiteOk); ASSERT_NE(interpreter, nullptr); - ASSERT_EQ(interpreter->tensors_size(), 1); + ASSERT_EQ(interpreter->tensors_size(), 2); TfLiteTensor* t1 = interpreter->tensor(0); ASSERT_EQ(t1->allocation_type, kTfLiteMmapRo); diff --git a/tensorflow/lite/testdata/sparse_tensor.bin b/tensorflow/lite/testdata/sparse_tensor.bin index c035e02441d..ef023280887 100644 Binary files a/tensorflow/lite/testdata/sparse_tensor.bin and b/tensorflow/lite/testdata/sparse_tensor.bin differ diff --git a/tensorflow/lite/testdata/sparse_tensor.json b/tensorflow/lite/testdata/sparse_tensor.json index d23c0d0a64b..3c6a742a4e8 100644 --- a/tensorflow/lite/testdata/sparse_tensor.json +++ b/tensorflow/lite/testdata/sparse_tensor.json @@ -40,11 +40,19 @@ } ] } + }, + { + "shape": [ + 4, + 4 + ], + "name": "output_tensor", + "type": "INT8" } ], "inputs": [0], "outputs": [0], - "operators": [{"inputs":[-1], "outputs":[-1]}] + "operators": [{"inputs":[-1], "outputs":[1]}] } ], "buffers": [