Commit e3002e99 authored by Sebastian Nickels's avatar Sebastian Nickels

Updated LayerNameCreator

parent fa17c1ff
Pipeline #180583 failed with stages
in 2 minutes and 39 seconds
...@@ -38,7 +38,11 @@ public class LayerNameCreator { ...@@ -38,7 +38,11 @@ public class LayerNameCreator {
stage = name(networkInstruction.getBody(), stage, new ArrayList<>()); stage = name(networkInstruction.getBody(), stage, new ArrayList<>());
if (networkInstruction.isUnroll()) { if (networkInstruction.isUnroll()) {
// TODO: Add individual timesteps? UnrollInstructionSymbol unroll = (UnrollInstructionSymbol) networkInstruction;
for (SerialCompositeElementSymbol body : unroll.getResolvedBodies()) {
stage = name(body, stage, new ArrayList<>());
}
} }
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment