// 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});
}