experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update sparse_tensor model.

Browse Source 
PiperOrigin-RevId: 301879775
Change-Id: I5a82a592f80f2af6624328a3596f27cf9a8c9959
This commit is contained in:
Yunlu Li 2020-03-19 13:07:34 -07:00 committed by TensorFlower Gardener
parent 111879a5bc
commit b65b240474
3 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/model_test.cc
View File

@ -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);

BIN
tensorflow/lite/testdata/sparse_tensor.bin vendored
View File

Binary file not shown.

10
tensorflow/lite/testdata/sparse_tensor.json vendored
View File

@ -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": [