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

[Add] First draft of a test for horizontalTiling transformation

parent 9cddbffc
No related branches found
No related tags found
No related merge requests found
/********************************************************************************
* Copyright (c) 2023 CEA-List
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* SPDX-License-Identifier: EPL-2.0
*
********************************************************************************/
#include <catch2/catch_test_macros.hpp>
#include <set>
#include "aidge/graph/GraphView.hpp"
#include "aidge/graph/OpArgs.hpp"
#include "aidge/operator/Conv.hpp"
#include "aidge/operator/ReLU.hpp"
#include "aidge/recipies/Recipies.hpp"
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;
}
}
} // 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