Skip to content
Snippets Groups Projects

Scheduler backprop

Merged Maxence Naud requested to merge scheduler_backprop into dev
1 file
+ 2
2
Compare changes
  • Side-by-side
  • Inline
@@ -210,7 +210,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") {
}
TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward]") {
std::shared_ptr<GraphView> gv = Sequential({ReLU(), ReLU()});
std::shared_ptr<GraphView> gv = Sequential({ReLU("relu0"), Sqrt("srqt0"), ReLU("relu1")});
std::shared_ptr<Tensor> inputTensor =
std::make_shared<Tensor>(Array4D<int, 2, 1, 5, 5>{{{{{0, 1, 2, 3, 4},
@@ -227,7 +227,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
// implem already set to default
auto myProd = Producer(inputTensor, "prod");
myProd -> addChild(gv);
gv -> compile("cpu", DataType::Float32);
gv -> compile("cpu", DataType::Int32);
compile_gradient(gv);
SequentialScheduler scheduler(gv);
scheduler.backward();
Loading