Commit 0b46c76e authored by aa's avatar aa Committed by Julian Johannes Steinsberger-Dührßen

Added Tensorflow generator

parent d8a9cb6f
......@@ -68,7 +68,7 @@ public enum Backend {
case "GLUON":
return Optional.of(GLUON);
case "TENSORFLOW":
return Optional.of(TENSORFLOW);
......@@ -81,10 +81,13 @@ public enum Backend {
switch (backend){
case CAFFE2:
return "CAFFE2";
case GLUON:
return "GLUON";
case TENSORFLOW:
return "TENSORFLOW";
default:
return "MXNET";
}
......
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