Skip to content
This repository has been archived by the owner on Oct 1, 2020. It is now read-only.

Commit

Permalink
Skip duplicate SqueezeNet operators in convolution microbenchmark
Browse files Browse the repository at this point in the history
  • Loading branch information
Marat Dukhan committed Dec 29, 2018
1 parent 1a950a2 commit bf17a75
Showing 1 changed file with 14 additions and 12 deletions.
26 changes: 14 additions & 12 deletions bench/convolution.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -409,6 +409,7 @@ static void MobileNetV2(benchmark::internal::Benchmark* b) {
b->Args({1, 1, 1, 1, 1, 1, 1, 1, 1280, 1000});
}

/* SqueezeNet 1.0 */
static void SqueezeNetV10(benchmark::internal::Benchmark* b) {
b->ArgNames({"N", "H", "W", "KH", "KW", "S", "D", "G", "GCin", "GCout"});

Expand All @@ -423,8 +424,8 @@ static void SqueezeNetV10(benchmark::internal::Benchmark* b) {
/********************** Fire 3 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 56, 55, 1, 1, 1, 1, 1, 128, 16});
b->Args({1, 55, 55, 1, 1, 1, 1, 1, 16, 64});
b->Args({1, 55, 55, 3, 3, 1, 1, 1, 16, 64});
/*b->Args({1, 55, 55, 1, 1, 1, 1, 1, 16, 64});*/
/*b->Args({1, 55, 55, 3, 3, 1, 1, 1, 16, 64});*/
/********************** Fire 4 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 55, 55, 1, 1, 1, 1, 1, 128, 32});
Expand All @@ -443,8 +444,8 @@ static void SqueezeNetV10(benchmark::internal::Benchmark* b) {
/********************** Fire 7 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 27, 27, 1, 1, 1, 1, 1, 384, 48});
b->Args({1, 27, 27, 1, 1, 1, 1, 1, 48, 192});
b->Args({1, 27, 27, 3, 3, 1, 1, 1, 48, 192});
/*b->Args({1, 27, 27, 1, 1, 1, 1, 1, 48, 192});*/
/*b->Args({1, 27, 27, 3, 3, 1, 1, 1, 48, 192});*/
/********************** Fire 8 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 27, 27, 1, 1, 1, 1, 1, 384, 64});
Expand All @@ -460,6 +461,7 @@ static void SqueezeNetV10(benchmark::internal::Benchmark* b) {
b->Args({1, 13, 13, 1, 1, 1, 1, 1, 512, 1000});
}

/* SqueezeNet 1.1 */
static void SqueezeNetV11(benchmark::internal::Benchmark* b) {
b->ArgNames({"N", "H", "W", "KH", "KW", "S", "D", "G", "GCin", "GCout"});

Expand All @@ -474,8 +476,8 @@ static void SqueezeNetV11(benchmark::internal::Benchmark* b) {
/********************** Fire 3 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 55, 55, 1, 1, 1, 1, 1, 128, 16});
b->Args({1, 55, 55, 1, 1, 1, 1, 1, 16, 64});
b->Args({1, 55, 55, 3, 3, 1, 1, 1, 16, 64});
/*b->Args({1, 55, 55, 1, 1, 1, 1, 1, 16, 64});*/
/*b->Args({1, 55, 55, 3, 3, 1, 1, 1, 16, 64});*/
/********************** Fire 4 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 27, 27, 1, 1, 1, 1, 1, 128, 32});
Expand All @@ -484,8 +486,8 @@ static void SqueezeNetV11(benchmark::internal::Benchmark* b) {
/********************** Fire 5 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 27, 27, 1, 1, 1, 1, 1, 256, 32});
b->Args({1, 27, 27, 1, 1, 1, 1, 1, 32, 128});
b->Args({1, 27, 27, 3, 3, 1, 1, 1, 32, 128});
/*b->Args({1, 27, 27, 1, 1, 1, 1, 1, 32, 128});*/
/*b->Args({1, 27, 27, 3, 3, 1, 1, 1, 32, 128});*/
/********************** Fire 6 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 13, 13, 1, 1, 1, 1, 1, 256, 48});
Expand All @@ -494,8 +496,8 @@ static void SqueezeNetV11(benchmark::internal::Benchmark* b) {
/********************** Fire 7 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 13, 13, 1, 1, 1, 1, 1, 384, 48});
b->Args({1, 13, 13, 1, 1, 1, 1, 1, 48, 192});
b->Args({1, 13, 13, 3, 3, 1, 1, 1, 48, 192});
/*b->Args({1, 13, 13, 1, 1, 1, 1, 1, 48, 192});*/
/*b->Args({1, 13, 13, 3, 3, 1, 1, 1, 48, 192});*/
/********************** Fire 8 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 13, 13, 1, 1, 1, 1, 1, 384, 64});
Expand All @@ -504,8 +506,8 @@ static void SqueezeNetV11(benchmark::internal::Benchmark* b) {
/********************** Fire 9 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 13, 13, 1, 1, 1, 1, 1, 512, 64});
b->Args({1, 13, 13, 1, 1, 1, 1, 1, 64, 256});
b->Args({1, 13, 13, 3, 3, 1, 1, 1, 64, 256});
/*b->Args({1, 13, 13, 1, 1, 1, 1, 1, 64, 256});*/
/*b->Args({1, 13, 13, 3, 3, 1, 1, 1, 64, 256});*/
/********************* Conv 10 *********************/
/* N H W KH KW S D G GCin GCout */
b->Args({1, 13, 13, 1, 1, 1, 1, 1, 512, 1000});
Expand Down

0 comments on commit bf17a75

Please sign in to comment.