Internal infra change
PiperOrigin-RevId: 357744680 Change-Id: Iaaf3ac33c9ce032894577892eff947440a1f3f60
This commit is contained in:
parent
b8b03dc654
commit
c87f1e3c06
@ -38,11 +38,11 @@ public final class InterpreterMobileNetTest {
|
|||||||
|
|
||||||
private static final ByteBuffer MOBILENET_FLOAT_MODEL_BUFFER =
|
private static final ByteBuffer MOBILENET_FLOAT_MODEL_BUFFER =
|
||||||
TestUtils.getTestFileAsBuffer(
|
TestUtils.getTestFileAsBuffer(
|
||||||
"third_party/tensorflow/lite/java/demo/app/src/main/assets/mobilenet_v1_1.0_224.tflite");
|
"tensorflow/lite/java/demo/app/src/main/assets/mobilenet_v1_1.0_224.tflite");
|
||||||
|
|
||||||
private static final ByteBuffer MOBILENET_QUANTIZED_MODEL_BUFFER =
|
private static final ByteBuffer MOBILENET_QUANTIZED_MODEL_BUFFER =
|
||||||
TestUtils.getTestFileAsBuffer(
|
TestUtils.getTestFileAsBuffer(
|
||||||
"third_party/tensorflow/lite/java/demo/app/src/main/assets/mobilenet_v1_1.0_224_quant.tflite");
|
"tensorflow/lite/java/demo/app/src/main/assets/mobilenet_v1_1.0_224_quant.tflite");
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testMobileNet() {
|
public void testMobileNet() {
|
||||||
|
@ -39,7 +39,7 @@ public final class GpuDelegateTest {
|
|||||||
private static final ByteBuffer MODEL_BUFFER = TestUtils.getTestFileAsBuffer(MODEL_PATH);
|
private static final ByteBuffer MODEL_BUFFER = TestUtils.getTestFileAsBuffer(MODEL_PATH);
|
||||||
private static final ByteBuffer MOBILENET_QUANTIZED_MODEL_BUFFER =
|
private static final ByteBuffer MOBILENET_QUANTIZED_MODEL_BUFFER =
|
||||||
TestUtils.getTestFileAsBuffer(
|
TestUtils.getTestFileAsBuffer(
|
||||||
"third_party/tensorflow/lite/java/demo/app/src/main/assets/mobilenet_v1_1.0_224_quant.tflite");
|
"tensorflow/lite/java/demo/app/src/main/assets/mobilenet_v1_1.0_224_quant.tflite");
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testBasic() throws Exception {
|
public void testBasic() throws Exception {
|
||||||
|
@ -64,12 +64,12 @@ bool RunWithRandomInputs(const std::string& filename) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
TEST(SelectiveBuiltTest, AddModel) {
|
TEST(SelectiveBuiltTest, AddModel) {
|
||||||
std::string model = "third_party/tensorflow/lite/testdata/add.bin";
|
std::string model = "tensorflow/lite/testdata/add.bin";
|
||||||
EXPECT_THAT(RunWithRandomInputs(model), true);
|
EXPECT_THAT(RunWithRandomInputs(model), true);
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST(SelectiveBuiltTest, LSTMModel) {
|
TEST(SelectiveBuiltTest, LSTMModel) {
|
||||||
std::string model = "third_party/tensorflow/lite/testdata/lstm.bin";
|
std::string model = "tensorflow/lite/testdata/lstm.bin";
|
||||||
EXPECT_THAT(RunWithRandomInputs(model), true);
|
EXPECT_THAT(RunWithRandomInputs(model), true);
|
||||||
}
|
}
|
||||||
} // namespace tflite
|
} // namespace tflite
|
||||||
|
Loading…
x
Reference in New Issue
Block a user