Skip to content
Snippets Groups Projects
Commit 987c568c authored by Maxence Naud's avatar Maxence Naud
Browse files

[Upd] minor changes and comment horizontal tiling test file

parent 5fc8cad9
No related branches found
No related tags found
No related merge requests found
......@@ -48,9 +48,9 @@ public:
template <ScalingAttr e> using attr = typename Attributes_::template attr<e>;
Scaling_Op(float scalingFactor)
: Operator(Type),
Attributes_(
attr<ScalingAttr::scalingFactor>(scalingFactor))
: Operator(Type),
Attributes_(
attr<ScalingAttr::scalingFactor>(scalingFactor))
{
setDatatype(DataType::Float32);
}
......
......@@ -23,25 +23,26 @@ namespace Aidge {
TEST_CASE("[core/recipies] Tiling(horizontal)", "[Tiling][HorizontalTiling][Recipies]") {
SECTION("Transform a pre-generated GraphView") {
std::shared_ptr<GraphView> g = Sequential({
Conv(3, 16, {3,3}, "conv1"),
ReLU("relu1"),
Conv(16, 32, {1,1}, "conv2"),
Conv(32, 16, {1,1}, "conv3"),
Conv(16, 10, {3,3}, "conv4"),
ReLU("relu2")
});
for (auto& individualConv : g->match("Conv")) {
auto tiledConv = horizontalTiling(individualConv);
g->replace(individualConv, tiledConv);
}
}
SECTION("Create the GraphView with tiled layers") {
std::shared_ptr<GraphView> g;
}
// SECTION("Transform a pre-generated GraphView") {
// std::shared_ptr<GraphView> g = Sequential({
// Conv(3, 16, {3,3}, "conv1"),
// ReLU("relu1"),
// Conv(16, 32, {1,1}, "conv2"),
// Conv(32, 16, {1,1}, "conv3"),
// Conv(16, 10, {3,3}, "conv4"),
// ReLU("relu2")
// });
// for (auto& individualConv : g->match("Conv")) {
// auto tiledConv = horizontalTiling(individualConv);
// g->replace(individualConv, tiledConv);
// }
// }
// SECTION("Create the GraphView with tiled layers") {
// std::shared_ptr<GraphView> g;
// g->addChild(horizontalTiling(Conv()))
// }
}
} // namespace Aidge
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment