mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-05 17:34:21 +00:00
Migrate conv2d layer
This commit is contained in:
@@ -3,7 +3,7 @@
|
||||
|
||||
#include <iostream>
|
||||
|
||||
#include "conv2d.cuh"
|
||||
#include "conv2d.hpp"
|
||||
|
||||
class Conv2dTest : public ::testing::Test {
|
||||
protected:
|
||||
|
||||
@@ -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) {
|
||||
|
||||
Reference in New Issue
Block a user