Implement getOutputSize and getInputSize for seq layers

2025-11-06 09:44:28 +00:00 · 2024-04-22 20:31:58 +02:00
parent fbf6c44bdd
commit f17debc244
18 changed files with 186 additions and 66 deletions
--- a/test/layers/test_avg_pooling.cu
+++ b/test/layers/test_avg_pooling.cu
@@ -51,10 +51,10 @@ TEST(AvgPoolingLayerTest, AvgPoolForwardTest) {

    int outputSize = avgPoolingLayer.getOutputSize();

-    std::vector<float> output(outputSize * outputSize * nChannels);
+    std::vector<float> output(outputSize);
    cudaStatus = cudaMemcpy(
        output.data(), d_output,
-        sizeof(float) * outputSize * outputSize * nChannels,
+        sizeof(float) * outputSize,
        cudaMemcpyDeviceToHost
    );
    EXPECT_EQ(cudaStatus, cudaSuccess);