Commit 8df67e4a authored by David Tellenbach's avatar David Tellenbach
Browse files

Merge branch 'precision_learning_gpu' of gitlab.lrz.de:IP/elsa into precision_learning_gpu

parents fbf607f7 7c1d08f2
Pipeline #459563 failed with stages
in 41 minutes and 8 seconds
......@@ -33,7 +33,8 @@ TEMPLATE_TEST_CASE("ConvShapes", "[ml]", float)
auto input = ml::Input(inputDesc, 1);
auto layer = ml::Conv2D<real_t>(3, std::array<index_t, 3>({7, 7, 64}), ml::Activation::Relu, 3, ml::Padding::Same);
auto layer = ml::Conv2D<real_t>(3, std::array<index_t, 3>({7, 7, 64}), ml::Activation::Relu,
3, ml::Padding::Same);
layer.setInput(&input);
layer.setInputDescriptor(inputDesc);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment