Commit 0b271d6a authored by Julian Treiber's avatar Julian Treiber

added Crop Layer to supported layers

parent 1cfd7cf9
...@@ -40,6 +40,7 @@ public class CNNArch2GluonLayerSupportChecker extends LayerSupportChecker { ...@@ -40,6 +40,7 @@ public class CNNArch2GluonLayerSupportChecker extends LayerSupportChecker {
supportedLayerList.add(AllPredefinedLayers.REDUCE_SUM_NAME); supportedLayerList.add(AllPredefinedLayers.REDUCE_SUM_NAME);
supportedLayerList.add(AllPredefinedLayers.BROADCAST_ADD_NAME); supportedLayerList.add(AllPredefinedLayers.BROADCAST_ADD_NAME);
supportedLayerList.add(AllPredefinedLayers.RESHAPE_NAME); supportedLayerList.add(AllPredefinedLayers.RESHAPE_NAME);
supportedLayerList.add(AllPredefinedLayers.CROP_NAME);
} }
} }
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