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