diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-10-25 19:17:35 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:55:45 +0000 |
commit | 9593bde92b99fb4473611a2a2bb47a8040bfb500 (patch) | |
tree | 82f869b95d6b79c2e19880e3f0d22a4eb31c0c8a /tests/benchmark/DragonBenchConfigs.h | |
parent | 89d71731d8922bc302ac57046126cdaedcf6e96b (diff) | |
download | ComputeLibrary-9593bde92b99fb4473611a2a2bb47a8040bfb500.tar.gz |
COMPMID-1674: (3RDPARTY_UPDATE) Add FullyConnectedLayer Dragonbench
Change-Id: I3c3e96a743614af4c2c2391780d5de2db6191b0f
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/155318
Tested-by: bsgcomp <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'tests/benchmark/DragonBenchConfigs.h')
-rw-r--r-- | tests/benchmark/DragonBenchConfigs.h | 54 |
1 files changed, 53 insertions, 1 deletions
diff --git a/tests/benchmark/DragonBenchConfigs.h b/tests/benchmark/DragonBenchConfigs.h index c12a22c8ae..4fe6c79ae8 100644 --- a/tests/benchmark/DragonBenchConfigs.h +++ b/tests/benchmark/DragonBenchConfigs.h @@ -28,10 +28,59 @@ #include <string> #include <vector> -#include "dragonbench/conv2d.hpp" +#include "dragonbench/conv2d/conv2d.hpp" +#include "dragonbench/fully_connected/fully_connected.hpp" namespace arm_compute { +// Stream operators +inline ::std::ostream &operator<<(::std::ostream &os, const Conv2D &conv2d_config) +{ + os << "network_name=" << conv2d_config.network_name << ":"; + os << "layer_name=" << conv2d_config.layer_name << ":"; + os << "id=" << conv2d_config.id << ":"; + os << "Input_NCHW=" + << conv2d_config.ibatch << ',' + << conv2d_config.ch_in << ',' + << conv2d_config.dim_in_h << ',' + << conv2d_config.dim_in_w << ":"; + os << "Output_NCHW=" + << conv2d_config.ibatch << ',' + << conv2d_config.ch_out << ',' + << conv2d_config.dim_out_h << ',' + << conv2d_config.dim_out_w << ":"; + os << "Weights_HW=" + << conv2d_config.kern_h << ',' + << conv2d_config.kern_w << ":"; + os << "Stride_HW=" + << conv2d_config.stride_h << ',' + << conv2d_config.stride_w << ":"; + os << "Padding=" << conv2d_config.padding << ":"; + return os; +} +inline std::string to_string(const Conv2D &conv2d_config) +{ + std::stringstream str; + str << conv2d_config; + return str.str(); +} +inline ::std::ostream &operator<<(::std::ostream &os, const Fully_Connected &fc_config) +{ + os << "network_name=" << fc_config.network_name << ":"; + os << "layer_name=" << fc_config.layer_name << ":"; + os << "id=" << fc_config.id << ":"; + os << "M=" << fc_config.m << ":"; + os << "N=" << fc_config.n << ":"; + os << "K=" << fc_config.k << ":"; + return os; +} +inline std::string to_string(const Fully_Connected &fc_config) +{ + std::stringstream str; + str << fc_config; + return str.str(); +} + namespace test { namespace benchmark @@ -40,6 +89,9 @@ namespace benchmark extern conv2d_configs silverwing_cfgs; extern conv2d_configs sunfyre_cfgs; extern conv2d_configs syrax_cfgs; + +// Fully Connected benchmarks +extern fully_connected_configs dreamfyre_cfgs; } // namespace benchmark } // namespace test } // namespace arm_compute |