test / bench /conv.h
Androidonnxfork's picture
Upload folder using huggingface_hub
8b7c501
raw
history blame
45.7 kB
// Copyright (c) Facebook, Inc. and its affiliates.
// All rights reserved.
//
// Copyright 2019 Google LLC
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.
#pragma once
#include <benchmark/benchmark.h>
#define BENCHMARK_CONV(conv_fn) \
BENCHMARK_CAPTURE(conv_fn, mobilenet_v1, "MobileNet v1")->Apply(MobileNetV1ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, mobilenet_v2, "MobileNet v2")->Apply(MobileNetV2ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, mobilenet_v3_small, "MobileNet v3 Small")->Apply(MobileNetV3SmallConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, mobilenet_v3_large, "MobileNet v3 Large")->Apply(MobileNetV3LargeConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, shufflenet_v1_g1, "ShuffleNet v1 (1 group)")->Apply(ShuffleNetV1G1ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, shufflenet_v1_g2, "ShuffleNet v1 (2 groups)")->Apply(ShuffleNetV1G2ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, shufflenet_v1_g3, "ShuffleNet v1 (3 groups)")->Apply(ShuffleNetV1G3ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, shufflenet_v1_g4, "ShuffleNet v1 (4 groups)")->Apply(ShuffleNetV1G4ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, shufflenet_v1_g8, "ShuffleNet v1 (8 groups)")->Apply(ShuffleNetV1G8ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, shufflenet_v2_x05, "ShuffleNet v2 0.5X")->Apply(ShuffleNetV2X05ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, shufflenet_v2_x10, "ShuffleNet v2 1.0X")->Apply(ShuffleNetV2X10ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, shufflenet_v2_x15, "ShuffleNet v2 1.5X")->Apply(ShuffleNetV2X15ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, shufflenet_v2_x20, "ShuffleNet v2 2.0X")->Apply(ShuffleNetV2X20ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, inception_v3, "Inception v3")->Apply(InceptionV3ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, resnet18, "ResNet-18")->Apply(ResNet18ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, resnet50, "ResNet-50")->Apply(ResNet50ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, squeezenet_v10, "SqueezeNet 1.0")->Apply(SqueezeNetV10ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, squeezenet_v11, "SqueezeNet 1.1")->Apply(SqueezeNetV11ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, vgg, "VGG")->Apply(VGGConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, srcnn915, "SRCNN (9-1-5)")->Apply(SRCNN915ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, srcnn935, "SRCNN (9-3-5)")->Apply(SRCNN935ConvArguments)->UseRealTime(); \
BENCHMARK_CAPTURE(conv_fn, srcnn955, "SRCNN (9-5-5)")->Apply(SRCNN955ConvArguments)->UseRealTime();
// ShuffleNet v1 with 1 group.
static void ShuffleNetV1G1ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 24});
/*************** Stage 2: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 36});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 36, 120});
/*************** Stage 2: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 144, 36});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 36, 144});
/*************** Stage 3: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 144, 72});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 72, 144});
/*************** Stage 3: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 288, 72});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 72, 288});
/*************** Stage 4: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 288, 144});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 144, 288});
/*************** Stage 4: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 576, 144});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 144, 576});
}
// ShuffleNet v1 with 2 groups.
static void ShuffleNetV1G2ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 24});
/*************** Stage 2: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 50});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 25, 88});
/*************** Stage 2: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 100, 25});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 25, 100});
/*************** Stage 3: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 100, 50});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 50, 100});
/*************** Stage 3: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 200, 50});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 50, 200});
/*************** Stage 4: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 200, 100});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 100, 200});
/*************** Stage 4: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 400, 100});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 100, 400});
}
// ShuffleNet v1 with 3 groups.
static void ShuffleNetV1G3ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 24});
/*************** Stage 2: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 60});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 20, 72});
/*************** Stage 2: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 80, 20});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 20, 80});
/*************** Stage 3: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 80, 40});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 40, 80});
/*************** Stage 3: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 160, 40});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 40, 160});
/*************** Stage 4: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 160, 80});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 80, 160});
/*************** Stage 4: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 320, 80});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 80, 320});
}
// ShuffleNet v1 with 4 groups.
static void ShuffleNetV1G4ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 24});
/*************** Stage 2: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 68});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 17, 62});
/*************** Stage 2: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 68, 17});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 17, 68});
/*************** Stage 3: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 68, 34});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 34, 68});
/*************** Stage 3: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 136, 34});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 34, 136});
/*************** Stage 4: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 136, 68});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 68, 136});
/*************** Stage 4: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 272, 68});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 68, 272});
}
// ShuffleNet v1 with 8 groups.
static void ShuffleNetV1G8ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 24});
/*************** Stage 2: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 96});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 12, 45});
/*************** Stage 2: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 48, 12});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 12, 48});
/*************** Stage 3: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 48, 24});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 24, 48});
/*************** Stage 3: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 96, 24});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 24, 96});
/*************** Stage 4: stride-2 unit **************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 96, 48});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 48, 96});
/*************** Stage 4: stride-1 units *************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 192, 48});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 48, 192});
}
// ShuffleNet v2 (0.5X scale).
static void ShuffleNetV2X05ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 24});
/********************** Stage 2 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 24, 24});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 24});
/********************** Stage 3 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 48, 48});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 48, 48});
/********************** Stage 4 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 96, 96});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 96, 96});
/*********************** Conv 5 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 192, 1024});
}
// ShuffleNet v2 (1.0X scale).
static void ShuffleNetV2X10ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 24});
/********************** Stage 2 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 24, 58});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 58});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 58, 58});
/********************** Stage 3 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 116, 116});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 116, 116});
/********************** Stage 4 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 232, 232});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 232, 232});
/*********************** Conv 5 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 464, 1024});
}
// ShuffleNet v2 (1.5X scale).
static void ShuffleNetV2X15ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 24});
/********************** Stage 2 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 24, 88});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 88});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 88, 88});
/********************** Stage 3 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 176, 176});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 176, 176});
/********************** Stage 4 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 352, 352});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 352, 352});
/*********************** Conv 5 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 704, 1024});
}
// ShuffleNet v2 (2.0X scale).
static void ShuffleNetV2X20ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 24});
/********************** Stage 2 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 24, 122});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 122});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 122, 122});
/********************** Stage 3 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 244, 244});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 244, 244});
/********************** Stage 4 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 488, 488});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 488, 488});
/*********************** Conv 5 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 976, 2048});
}
static void MobileNetV1ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 32});
b->Args({112, 112, 1, 1, 0, 0, 1, 1, 32, 64});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 64, 128});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 128, 128});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 128, 256});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 256, 256});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 256, 512});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 512, 512});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 512, 1024});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 1024, 1024});
}
static void MobileNetV2ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 32});
/******************** Bottleneck 1 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({112, 112, 1, 1, 0, 0, 1, 1, 32, 16});
/******************** Bottleneck 2 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({112, 112, 1, 1, 0, 0, 1, 1, 16, 96});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 96, 24});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 144});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 144, 24});
/******************** Bottleneck 3 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 144});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 144, 32});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 32, 192});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 192, 32});
//b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 32, 192});
//b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 192, 32});
/******************** Bottleneck 4 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 32, 192});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 192, 64});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 64, 384});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 384, 64});
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 64, 384});
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 384, 64});
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 64, 384});
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 384, 64});
/******************** Bottleneck 5 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 64, 384});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 384, 96});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 96, 576});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 576, 96});
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 96, 576});
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 576, 96});
/******************** Bottleneck 6 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 96, 576});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 576, 160});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 160, 960});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 960, 160});
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 160, 960});
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 960, 160});
/******************** Bottleneck 7 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 160, 960});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 960, 320});
/**************** Pre-pooling Conv2D *****************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 320, 1280});
/**************** Post-pooling Conv2D ****************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 1280, 1000});
}
static void MobileNetV3SmallConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/******************* Initial Stage *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 16});
/******************** Bottleneck 1 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 16, 8});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 8, 16});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 16, 16});
/******************** Bottleneck 2 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 16, 72});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 72, 24});
/******************** Bottleneck 3 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 24, 88});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 88, 24});
/******************** Bottleneck 4 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 24, 96});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 96, 24});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 24, 96});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 96, 40});
/******************** Bottleneck 5 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 40, 240});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 240, 64});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 64, 240});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 240, 40});
/******************** Bottleneck 6 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 40, 240});
//b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 240, 64});
//b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 64, 240});
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 240, 40});
/******************** Bottleneck 7 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 40, 120});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 120, 32});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 32, 120});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 120, 48});
/******************** Bottleneck 8 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 48, 144});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 144, 40});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 40, 144});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 144, 48});
/******************** Bottleneck 9 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 48, 288});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 288, 72});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 72, 288});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 288, 96});
/******************* Bottleneck 10 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 96, 576});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 576, 144});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 144, 576});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 576, 96});
/******************* Bottleneck 11 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 96, 576});
//b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 576, 144});
//b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 144, 576});
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 576, 96});
/********************* Last Stage ********************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 96, 576});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 576, 1024});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 1024, 1001});
}
static void MobileNetV3LargeConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/******************* Initial Stage *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 16});
/******************** Bottleneck 1 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({112, 112, 1, 1, 0, 0, 1, 1, 16, 16});
/******************** Bottleneck 2 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({112, 112, 1, 1, 0, 0, 1, 1, 16, 64});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 64, 24});
/******************** Bottleneck 3 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 72});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 72, 24});
/******************** Bottleneck 4 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 24, 72});*/
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 72, 24});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 24, 72});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 72, 40});
/******************** Bottleneck 5 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 40, 120});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 120, 32});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 32, 120});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 120, 40});
/******************** Bottleneck 6 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 40, 120});
//b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 120, 32});
//b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 32, 120});
//b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 120, 40});
/******************** Bottleneck 7 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 40, 240});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 240, 80});
/******************** Bottleneck 8 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 80, 200});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 200, 80});
/******************** Bottleneck 9 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 80, 184});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 184, 80});
/******************* Bottleneck 10 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 80, 184});
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 184, 80});
/******************* Bottleneck 11 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 80, 480});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 480, 120});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 120, 480});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 480, 112});
/******************* Bottleneck 12 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 112, 672});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 672, 168});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 168, 672});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 672, 112});
/******************* Bottleneck 13 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 112, 672});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 672, 160});
/******************* Bottleneck 14 *******************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 160, 960});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 960, 240});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 240, 960});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 960, 160});
/******************* Bottleneck 15 *******************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 160, 960});
//b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 960, 240});
//b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 240, 960});
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 960, 160});
/******************** Last Stage *********************/
/* H W KH KW PH PW S D GCin GCout */
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 160, 960});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 960, 1280});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 1280, 1001});
}
// SqueezeNet 1.0
static void SqueezeNetV10ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 7, 7, 6, 6, 2, 1, 3, 96});
/*********************** Fire 2 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 55, 55, 1, 1, 0, 0, 1, 1, 96, 16});
b->Args({ 55, 55, 1, 1, 0, 0, 1, 1, 16, 64});
b->Args({ 55, 55, 3, 3, 2, 2, 1, 1, 16, 64});
/*********************** Fire 3 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 55, 1, 1, 0, 0, 1, 1, 128, 16});
//b->Args({ 55, 55, 1, 1, 0, 0, 1, 1, 16, 64});
//b->Args({ 55, 55, 3, 3, 2, 2, 1, 1, 16, 64});
/*********************** Fire 4 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 55, 55, 1, 1, 0, 0, 1, 1, 128, 32});
b->Args({ 55, 55, 1, 1, 0, 0, 1, 1, 32, 128});
b->Args({ 55, 55, 3, 3, 2, 2, 1, 1, 32, 128});
/*********************** Fire 5 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 256, 32});
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 32, 128});
b->Args({ 27, 27, 3, 3, 2, 2, 1, 1, 32, 128});
/*********************** Fire 6 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 256, 48});
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 48, 192});
b->Args({ 27, 27, 3, 3, 2, 2, 1, 1, 48, 192});
/*********************** Fire 7 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 384, 48});
//b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 48, 192});
//b->Args({ 27, 27, 3, 3, 2, 2, 1, 1, 48, 192});
/*********************** Fire 8 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 384, 64});
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 64, 256});
b->Args({ 27, 27, 3, 3, 2, 2, 1, 1, 64, 256});
/*********************** Fire 9 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 512, 64});
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 64, 256});
b->Args({ 13, 13, 3, 3, 2, 2, 1, 1, 64, 256});
/********************** Conv 10 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 512, 1000});
}
// SqueezeNet 1.1
static void SqueezeNetV11ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/*********************** Conv 1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 2, 1, 3, 64});
/*********************** Fire 2 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 55, 55, 1, 1, 0, 0, 1, 1, 64, 16});
b->Args({ 55, 55, 1, 1, 0, 0, 1, 1, 16, 64});
b->Args({ 55, 55, 3, 3, 2, 2, 1, 1, 16, 64});
/*********************** Fire 3 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 55, 55, 1, 1, 0, 0, 1, 1, 128, 16});
//b->Args({ 55, 55, 1, 1, 0, 0, 1, 1, 16, 64});
//b->Args({ 55, 55, 3, 3, 2, 2, 1, 1, 16, 64});
/*********************** Fire 4 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 128, 32});
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 32, 128});
b->Args({ 27, 27, 3, 3, 2, 2, 1, 1, 32, 128});
/*********************** Fire 5 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 256, 32});
//b->Args({ 27, 27, 1, 1, 0, 0, 1, 1, 32, 128});
//b->Args({ 27, 27, 3, 3, 2, 2, 1, 1, 32, 128});
/*********************** Fire 6 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 256, 48});
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 48, 192});
b->Args({ 13, 13, 3, 3, 2, 2, 1, 1, 48, 192});
/*********************** Fire 7 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 384, 48});
//b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 48, 192});
//b->Args({ 13, 13, 3, 3, 2, 2, 1, 1, 48, 192});
/*********************** Fire 8 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 384, 64});
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 64, 256});
b->Args({ 13, 13, 3, 3, 2, 2, 1, 1, 64, 256});
/*********************** Fire 9 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 512, 64});
//b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 64, 256});
//b->Args({ 13, 13, 3, 3, 2, 2, 1, 1, 64, 256});
/********************** Conv 10 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 13, 13, 1, 1, 0, 0, 1, 1, 512, 1000});
}
static void InceptionV3ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/* H W KH KW PH PW S D GCin GCout */
b->Args({299, 299, 3, 3, 0, 0, 2, 1, 3, 32});
b->Args({149, 149, 3, 3, 0, 0, 1, 1, 32, 32});
b->Args({147, 147, 3, 3, 2, 2, 1, 1, 32, 64});
b->Args({ 73, 73, 1, 1, 0, 0, 1, 1, 64, 80});
b->Args({ 73, 73, 3, 3, 0, 0, 1, 1, 80, 192});
b->Args({ 35, 35, 1, 1, 0, 0, 1, 1, 192, 64});
b->Args({ 35, 35, 1, 1, 0, 0, 1, 1, 192, 48});
b->Args({ 35, 35, 5, 5, 4, 4, 1, 1, 48, 64});
b->Args({ 35, 35, 3, 3, 2, 2, 1, 1, 64, 96});
b->Args({ 35, 35, 3, 3, 2, 2, 1, 1, 96, 96});
b->Args({ 35, 35, 1, 1, 0, 0, 1, 1, 192, 32});
b->Args({ 35, 35, 1, 1, 0, 0, 1, 1, 256, 64});
b->Args({ 35, 35, 1, 1, 0, 0, 1, 1, 256, 48});
b->Args({ 35, 35, 1, 1, 0, 0, 1, 1, 288, 64});
b->Args({ 35, 35, 1, 1, 0, 0, 1, 1, 288, 48});
b->Args({ 35, 35, 3, 3, 0, 0, 2, 1, 288, 384});
b->Args({ 35, 35, 3, 3, 0, 0, 2, 1, 96, 96});
b->Args({ 17, 17, 1, 1, 0, 0, 1, 1, 768, 192});
b->Args({ 17, 17, 1, 1, 0, 0, 1, 1, 768, 128});
b->Args({ 17, 17, 1, 7, 0, 6, 1, 1, 128, 128});
b->Args({ 17, 17, 7, 1, 6, 0, 1, 1, 128, 192});
b->Args({ 17, 17, 7, 1, 6, 0, 1, 1, 128, 128});
b->Args({ 17, 17, 1, 7, 0, 6, 1, 1, 128, 192});
b->Args({ 17, 17, 1, 1, 0, 0, 1, 1, 768, 160});
b->Args({ 17, 17, 1, 7, 0, 6, 1, 1, 160, 160});
b->Args({ 17, 17, 7, 1, 6, 0, 1, 1, 160, 192});
b->Args({ 17, 17, 7, 1, 6, 0, 1, 1, 160, 160});
b->Args({ 17, 17, 1, 7, 0, 6, 1, 1, 160, 192});
b->Args({ 17, 17, 1, 7, 0, 6, 1, 1, 192, 192});
b->Args({ 17, 17, 7, 1, 6, 0, 1, 1, 192, 192});
b->Args({ 17, 17, 3, 3, 0, 0, 2, 1, 192, 320});
b->Args({ 17, 17, 3, 3, 0, 0, 2, 1, 192, 192});
b->Args({ 8, 8, 1, 1, 0, 0, 1, 1, 1280, 320});
b->Args({ 8, 8, 1, 1, 0, 0, 1, 1, 1280, 384});
b->Args({ 8, 8, 1, 3, 0, 2, 1, 1, 384, 384});
b->Args({ 8, 8, 3, 1, 2, 0, 1, 1, 384, 384});
b->Args({ 8, 8, 1, 1, 0, 0, 1, 1, 1280, 448});
b->Args({ 8, 8, 3, 3, 2, 2, 1, 1, 448, 384});
b->Args({ 8, 8, 1, 1, 0, 0, 1, 1, 1280, 192});
b->Args({ 8, 8, 1, 1, 0, 0, 1, 1, 2048, 320});
b->Args({ 8, 8, 1, 1, 0, 0, 1, 1, 2048, 384});
b->Args({ 8, 8, 1, 1, 0, 0, 1, 1, 2048, 448});
b->Args({ 8, 8, 1, 1, 0, 0, 1, 1, 2048, 192});
b->Args({ 1, 1, 1, 1, 0, 0, 1, 1, 2048, 1001});
}
static void ResNet18ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/********************** Conv 1 ***********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 7, 7, 6, 6, 2, 1, 3, 64});
/********************* Conv 2.X **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 3, 3, 2, 2, 1, 1, 64, 64});
/********************* Conv 3.X **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 3, 3, 2, 2, 2, 1, 64, 128});
b->Args({ 28, 28, 3, 3, 2, 2, 1, 1, 128, 128});
b->Args({ 56, 56, 1, 1, 0, 0, 2, 1, 64, 128});
/********************* Conv 4.X **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 3, 3, 2, 2, 2, 1, 128, 256});
b->Args({ 14, 14, 3, 3, 2, 2, 1, 1, 256, 256});
b->Args({ 28, 28, 1, 1, 0, 0, 2, 1, 128, 256});
/********************* Conv 5.X **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 3, 3, 2, 2, 2, 1, 256, 512});
b->Args({ 7, 7, 3, 3, 2, 2, 1, 1, 512, 512});
b->Args({ 14, 14, 1, 1, 0, 0, 2, 1, 256, 512});
}
static void ResNet50ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/********************** Conv 1 ***********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 7, 7, 6, 6, 2, 1, 3, 64});
/********************* Conv 2.1 **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 64, 64});
b->Args({ 56, 56, 3, 3, 2, 2, 1, 1, 64, 64});
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 64, 256});
//b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 64, 256});
/********************* Conv 2.X **********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 256, 64});
//b->Args({ 56, 56, 3, 3, 2, 2, 1, 1, 64, 64});
//b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 64, 256});
/********************** Conv 3.1 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 256, 128});
b->Args({ 56, 56, 3, 3, 2, 2, 2, 1, 128, 128});
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 128, 512});
b->Args({ 56, 56, 1, 1, 0, 0, 2, 1, 256, 512});
/********************** Conv 3.X *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 512, 128});
b->Args({ 28, 28, 3, 3, 2, 2, 1, 1, 128, 128});
//b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 128, 512});
/********************** Conv 4.1 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 512, 256});
b->Args({ 28, 28, 3, 3, 2, 2, 2, 1, 256, 256});
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 256, 1024});
b->Args({ 28, 28, 1, 1, 0, 0, 2, 1, 512, 1024});
/********************** Conv 4.X *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 1024, 256});
b->Args({ 14, 14, 3, 3, 2, 2, 1, 1, 256, 256});
//b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 256, 1024});
/********************** Conv 5.1 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 1024, 512});
b->Args({ 14, 14, 3, 3, 2, 2, 2, 1, 512, 512});
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 512, 2048});
b->Args({ 14, 14, 1, 1, 0, 0, 2, 1, 1024, 2048});
/********************** Conv 5.X *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 2048, 512});
b->Args({ 7, 7, 3, 3, 2, 2, 1, 1, 512, 512});
//b->Args({ 7, 7, 1, 1, 0, 0, 1, 1, 512, 2048});
}
static void VGGConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/********************** Conv 1.1 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 1, 1, 3, 64});
/********************** Conv 1.2 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({224, 224, 3, 3, 2, 2, 1, 1, 64, 64});
/********************** Conv 2.1 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({112, 112, 3, 3, 2, 2, 1, 1, 64, 128});
/********************** Conv 2.2 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({112, 112, 3, 3, 2, 2, 1, 1, 128, 128});
/********************** Conv 3.1 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 3, 3, 2, 2, 1, 1, 128, 256});
/********************** Conv 3.2 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 3, 3, 2, 2, 1, 1, 256, 256});
/********************** Conv 3.3 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 56, 56, 1, 1, 0, 0, 1, 1, 256, 256});
/********************** Conv 4.1 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 3, 3, 2, 2, 1, 1, 256, 512});
/********************** Conv 4.2 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 3, 3, 2, 2, 1, 1, 512, 512});
/********************** Conv 4.3 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 28, 28, 1, 1, 0, 0, 1, 1, 512, 512});
/********************** Conv 5.X *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 3, 3, 2, 2, 1, 1, 512, 512});
/********************** Conv 5.3 *********************/
/* H W KH KW PH PW S D GCin GCout */
b->Args({ 14, 14, 1, 1, 0, 0, 1, 1, 512, 512});
}
// SRCNN (9-1-5)
static void SRCNN915ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/* H W KH KW PH PW S D GCin GCout */
b->Args({384, 384, 9, 9, 0, 0, 1, 1, 1, 64});
b->Args({376, 376, 1, 1, 0, 0, 1, 1, 64, 32});
b->Args({376, 376, 5, 5, 0, 0, 1, 1, 32, 1});
}
// SRCNN (9-3-5)
static void SRCNN935ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/* H W KH KW PH PW S D GCin GCout */
b->Args({384, 384, 9, 9, 0, 0, 1, 1, 1, 64});
b->Args({376, 376, 3, 3, 0, 0, 1, 1, 64, 32});
b->Args({374, 374, 5, 5, 0, 0, 1, 1, 32, 1});
}
// SRCNN (9-5-5)
static void SRCNN955ConvArguments(benchmark::internal::Benchmark* b) {
b->ArgNames({"H", "W", "KH", "KW", "PH", "PW", "S", "D", "GCin", "GCout"});
/* H W KH KW PH PW S D GCin GCout */
b->Args({384, 384, 9, 9, 0, 0, 1, 1, 1, 64});
b->Args({376, 376, 5, 5, 0, 0, 1, 1, 64, 32});
b->Args({372, 372, 5, 5, 0, 0, 1, 1, 32, 1});
}