From b65b2404745f64d5aa38a9565c2f71a3fe52c0ca Mon Sep 17 00:00:00 2001 From: Yunlu Li <yunluli@google.com> Date: Thu, 19 Mar 2020 13:07:34 -0700 Subject: [PATCH] Update sparse_tensor model. PiperOrigin-RevId: 301879775 Change-Id: I5a82a592f80f2af6624328a3596f27cf9a8c9959 --- tensorflow/lite/model_test.cc | 2 +- tensorflow/lite/testdata/sparse_tensor.bin | Bin 508 -> 592 bytes tensorflow/lite/testdata/sparse_tensor.json | 10 +++++++++- 3 files changed, 10 insertions(+), 2 deletions(-) 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 c035e02441d68e6cb6dca9bfbbf47c2924335d0a..ef02328088720ccfb1946e9a798e306233bfd5eb 100644 GIT binary patch delta 196 zcmeyve1RoZg@J(~#LdSTNFf0}1`!4p1`Y-upqKyyACP1Q;y;WG3>H9|1&En|SOd!L z`uG1oNDQPJsDy<HsF;<3fepw7gOrJZ{(e9SkR-!@AOO)I0HZ;MMF6n?5c2>@4hD8G f3uGV=aLND)kOjOzoL^c}P+Bs1AEVgDBNmJRg+di@ delta 128 zcmcb>@`pK4f`Ne{#LdT;fq{=fgn@;DgMkMqA^_wuFfar07DfgJ10c-;#7sb}0A*kL u_y0dg45S*UgoO#Hn3aKnZOOz`f5rbm02E|^(Lg3h5g34IiOr&n){Fprk`vYd 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": [