Migrate conv2d layer

This commit is contained in:
2024-09-10 21:19:50 +02:00
parent 74f49d6a00
commit 757584544c
7 changed files with 208 additions and 163 deletions

View File

@@ -3,7 +3,7 @@
#include <iostream>
#include "conv2d.cuh"
#include "conv2d.hpp"
class Conv2dTest : public ::testing::Test {
protected:

View File

@@ -1,6 +1,6 @@
#include <gtest/gtest.h>
#include "conv2d.cuh"
#include "conv2d.hpp"
#include "dense.hpp"
#include "max_pooling.hpp"
#include "model.hpp"
@@ -85,8 +85,6 @@ class ModelTest : public ::testing::Test {
void commonTestTeardown(CUDANet::Model *model) {
delete model;
}
cudaError_t cudaStatus;
};
TEST_F(ModelTest, TestModelPredict) {