Commit d81f4106 authored by danielkisov's avatar danielkisov
Browse files

Update CNNArch2GluonLayerSupportChecker.java with custom layer that multiplies by one

parent cb20ed01
Pipeline #355017 failed with stage
in 1 minute and 30 seconds
......@@ -46,6 +46,7 @@ public class CNNArch2GluonLayerSupportChecker extends LayerSupportChecker {
supportedLayerList.add(AllPredefinedLayers.DOT_PRODUCT_SELF_ATTENTION_NAME);
supportedLayerList.add(AllPredefinedLayers.LOAD_NETWORK_NAME);
supportedLayerList.add(AllPredefinedLayers.LAYERNORM_NAME);
supportedLayerList.add(AllPredefinedLayers.MULTIPLY_BY_ONE);
}
}
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