Update sparse_tensor model.
PiperOrigin-RevId: 301879775 Change-Id: I5a82a592f80f2af6624328a3596f27cf9a8c9959
This commit is contained in:
parent
111879a5bc
commit
b65b240474
tensorflow/lite
@ -386,7 +386,7 @@ TEST(BasicFlatBufferModel, TestParseModelWithSparseTensor) {
|
|||||||
ASSERT_EQ(InterpreterBuilder(*model, TrivialResolver())(&interpreter),
|
ASSERT_EQ(InterpreterBuilder(*model, TrivialResolver())(&interpreter),
|
||||||
kTfLiteOk);
|
kTfLiteOk);
|
||||||
ASSERT_NE(interpreter, nullptr);
|
ASSERT_NE(interpreter, nullptr);
|
||||||
ASSERT_EQ(interpreter->tensors_size(), 1);
|
ASSERT_EQ(interpreter->tensors_size(), 2);
|
||||||
TfLiteTensor* t1 = interpreter->tensor(0);
|
TfLiteTensor* t1 = interpreter->tensor(0);
|
||||||
ASSERT_EQ(t1->allocation_type, kTfLiteMmapRo);
|
ASSERT_EQ(t1->allocation_type, kTfLiteMmapRo);
|
||||||
|
|
||||||
|
BIN
tensorflow/lite/testdata/sparse_tensor.bin
vendored
BIN
tensorflow/lite/testdata/sparse_tensor.bin
vendored
Binary file not shown.
10
tensorflow/lite/testdata/sparse_tensor.json
vendored
10
tensorflow/lite/testdata/sparse_tensor.json
vendored
@ -40,11 +40,19 @@
|
|||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"shape": [
|
||||||
|
4,
|
||||||
|
4
|
||||||
|
],
|
||||||
|
"name": "output_tensor",
|
||||||
|
"type": "INT8"
|
||||||
}
|
}
|
||||||
],
|
],
|
||||||
"inputs": [0],
|
"inputs": [0],
|
||||||
"outputs": [0],
|
"outputs": [0],
|
||||||
"operators": [{"inputs":[-1], "outputs":[-1]}]
|
"operators": [{"inputs":[-1], "outputs":[1]}]
|
||||||
}
|
}
|
||||||
],
|
],
|
||||||
"buffers": [
|
"buffers": [
|
||||||
|
Loading…
Reference in New Issue
Block a user