Match target names to the patterns in ml/ml-models
Change-Id: I4d112800f965c88d893c3cfd21a64ad4912d9611
diff --git a/benchmarks/BUILD b/benchmarks/BUILD
index 3de605d..56d1e8e 100644
--- a/benchmarks/BUILD
+++ b/benchmarks/BUILD
@@ -22,6 +22,41 @@
iss_test_size = "medium",
)
+kelvin_benchmark_simulator(
+ name = "mobilenet_v1_075_224_benchmark_simulator",
+ model = "//quant_models:mobilenet_v1_0.75_224_int8_dummy.tflite",
+ iterations = 3,
+ iss_test_size = "large",
+)
+
+kelvin_benchmark_simulator(
+ name = "mobilenet_v1_050_224_benchmark_simulator",
+ model = "//quant_models:mobilenet_v1_0.5_224_int8_dummy.tflite",
+ iterations = 3,
+ iss_test_size = "enormous",
+)
+
+kelvin_benchmark_simulator(
+ name = "mobilenet_v1_025_224_benchmark_simulator",
+ model = "//quant_models:mobilenet_v1_0.25_224_int8_dummy.tflite",
+ iterations = 3,
+ iss_test_size = "large",
+)
+
+kelvin_benchmark_simulator(
+ name = "mobilenet_v2_025_224_benchmark_simulator",
+ model = "//quant_models:mobilenet_v2_0.25_224_int8_dummy.tflite",
+ iterations = 3,
+ iss_test_size = "large",
+)
+
+kelvin_benchmark_simulator(
+ name = "mobilenet_v2_050_224_benchmark_simulator",
+ model = "//quant_models:mobilenet_v2_0.50_224_int8_dummy.tflite",
+ iterations = 3,
+ iss_test_size = "large",
+)
+
# Builds both ASIC and FPGA.
kelvin_benchmark_devices(
name = "person_detection_benchmark_device",
@@ -29,32 +64,32 @@
iterations = 2,
)
-kelvin_benchmark_simulator(
- name = "mobilenet_v1_075_224",
+kelvin_benchmark_devices(
+ name = "mobilenet_v1_075_224_benchmark_device",
model = "//quant_models:mobilenet_v1_0.75_224_int8_dummy.tflite",
iterations = 3,
)
-kelvin_benchmark_simulator(
- name = "mobilenet_v1_050_224",
+kelvin_benchmark_devices(
+ name = "mobilenet_v1_050_224_benchmark_device",
model = "//quant_models:mobilenet_v1_0.5_224_int8_dummy.tflite",
iterations = 3,
)
-kelvin_benchmark_simulator(
- name = "mobilenet_v1_025_224",
+kelvin_benchmark_devices(
+ name = "mobilenet_v1_025_224_benchmark_device",
model = "//quant_models:mobilenet_v1_0.25_224_int8_dummy.tflite",
iterations = 3,
)
-kelvin_benchmark_simulator(
- name = "mobilenet_v2_025_224",
+kelvin_benchmark_devices(
+ name = "mobilenet_v2_025_224_benchmark_device",
model = "//quant_models:mobilenet_v2_0.25_224_int8_dummy.tflite",
iterations = 3,
)
-kelvin_benchmark_simulator(
- name = "mobilenet_v2_050_224",
+kelvin_benchmark_devices(
+ name = "mobilenet_v2_050_224_benchmark_device",
model = "//quant_models:mobilenet_v2_0.50_224_int8_dummy.tflite",
iterations = 3,
-)
\ No newline at end of file
+)
