Skip to content
Snippets Groups Projects

Export refactor

Merged Cyril Moineau requested to merge allowNoInputProducer into dev
1 file
+ 2
1
Compare changes
  • Side-by-side
  • Inline
@@ -16,6 +16,7 @@ class test_scheduler(unittest.TestCase):
@@ -16,6 +16,7 @@ class test_scheduler(unittest.TestCase):
t = aidge_core.Tensor(np.arange(6, dtype=np.int32) - 3)
t = aidge_core.Tensor(np.arange(6, dtype=np.int32) - 3)
 
input_node = aidge_core.Producer(t)
relu = aidge_core.ReLU()
relu = aidge_core.ReLU()
input_node.add_child(relu)
input_node.add_child(relu)
@@ -29,7 +30,7 @@ class test_scheduler(unittest.TestCase):
@@ -29,7 +30,7 @@ class test_scheduler(unittest.TestCase):
scheduler = aidge_core.SequentialScheduler(gv)
scheduler = aidge_core.SequentialScheduler(gv)
scheduler.forward(data=[t])
scheduler.forward()
out_tensor = relu.get_operator().get_output(0)
out_tensor = relu.get_operator().get_output(0)
expected_out = [0,0,0,0,1,2]
expected_out = [0,0,0,0,1,2]
Loading