Skip to content
Snippets Groups Projects
Commit 25496756 authored by Jerome Hue's avatar Jerome Hue
Browse files

Cleanup files

parent 1b2b1699
No related branches found
No related tags found
2 merge requests!408[Add] Dropout Operator,!340Add: backward pass for Pop operator
Pipeline #65739 passed
......@@ -88,7 +88,7 @@ public:
void forward() override;
/**
* @brief Executes the forward pass for the `Pop` operation.
* @brief Executes the backward pass for the `Pop` operation.
*/
void backward() override;
};
......@@ -99,7 +99,7 @@ public:
*/
enum class PopAttr {
ForwardStep, // Tracks the current step in the forward pass
BackwardStep
BackwardStep // Tracks the current step in the backward pass
};
/**
......
......@@ -19,7 +19,7 @@
using namespace Aidge;
TEST_CASE("[cpu/operator] Pop(forward)", "[Pop][CPU]") {
TEST_CASE("[cpu/operator] Pop", "[Pop][CPU]") {
std::shared_ptr<Tensor> pop1 = std::make_shared<Tensor>(Array1D<int,3>{{4,5,6}});
std::shared_ptr<Tensor> pop2 = std::make_shared<Tensor>(Array1D<int,3>{{1,2,3}});
std::shared_ptr<Tensor> input = std::make_shared<Tensor>(Array2D<int,2,3>{{{1,2,3}, {4,5,6}}});
......@@ -35,7 +35,6 @@ TEST_CASE("[cpu/operator] Pop(forward)", "[Pop][CPU]") {
REQUIRE(*op->getOutput(0) == *pop2);
REQUIRE_NOTHROW(pop->forward());
REQUIRE(*op->getOutput(0) == *pop1);
// Backward
......
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