Commit 349d487d authored by Sebastian Nickels's avatar Sebastian Nickels
Browse files

Changed some layers to trainable

parent 54ec67ae
Pipeline #200228 passed with stages
in 17 minutes and 52 seconds
......@@ -34,11 +34,6 @@ public class OneHot extends PredefinedLayerDeclaration {
super(AllPredefinedLayers.ONE_HOT_NAME);
}
@Override
public boolean isTrainable() {
return false;
}
@Override
public List<ArchTypeSymbol> computeOutputTypes(List<ArchTypeSymbol> inputTypes, LayerSymbol layer, VariableSymbol.Member member) {
return Collections.singletonList(new ArchTypeSymbol.Builder()
......
......@@ -34,11 +34,6 @@ public class Split extends PredefinedLayerDeclaration {
super(AllPredefinedLayers.SPLIT_NAME);
}
@Override
public boolean isTrainable() {
return false;
}
@Override
public List<ArchTypeSymbol> computeOutputTypes(List<ArchTypeSymbol> inputTypes, LayerSymbol layer, VariableSymbol.Member member) {
ArchTypeSymbol inputShape = layer.getInputTypes().get(0);
......
......@@ -35,11 +35,6 @@ public class SwapAxes extends PredefinedLayerDeclaration {
super(AllPredefinedLayers.SWAPAXES_NAME);
}
@Override
public boolean isTrainable() {
return false;
}
@Override
public List<ArchTypeSymbol> computeOutputTypes(List<ArchTypeSymbol> inputTypes, LayerSymbol layer, VariableSymbol.Member member) {
......
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