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 +)