Python MKLDNN examples for CNN(LeNet) and RNN(LSTM) (#2932)

dding3 · Oct 28, 2019 · 5abb1e2 · 5abb1e2
1 parent ad39cf2
commit 5abb1e2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/...a/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/utils/python/api/PythonBigDL.scala b/...a/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/utils/python/api/PythonBigDL.scala
@@ -20,7 +20,7 @@ import java.util.{ArrayList => JArrayList, HashMap => JHashMap, List => JList, M
 
 import com.intel.analytics.bigdl._
 import com.intel.analytics.bigdl.dataset.{Identity => DIdentity, Sample => JSample, _}
-import com.intel.analytics.bigdl.nn.{PGCriterion, Zeros, _}
+import com.intel.analytics.bigdl.nn.{PGCriterion, Sequential, Zeros, _}
 import com.intel.analytics.bigdl.nn.abstractnn.{AbstractModule, _}
 import com.intel.analytics.bigdl.numeric._
 import com.intel.analytics.bigdl.optim.{Optimizer, _}
@@ -275,6 +275,10 @@ class PythonBigDL[T: ClassTag](implicit ev: TensorNumeric[T]) extends Serializab
       Sequential[T]()
   }
 
+  def toGraph(sequential: Sequential[T]): StaticGraph[T] = {
+    sequential.toGraph().asInstanceOf[StaticGraph[T]]
+  }
+
   def createAttention(hiddenSize: Int, numHeads: Int, attentionDropout: Float): Attention[T] = {
      Attention(hiddenSize, numHeads, attentionDropout)
   }
@@ -2504,6 +2508,10 @@ class PythonBigDL[T: ClassTag](implicit ev: TensorNumeric[T]) extends Serializab
     Engine.init
   }
 
+  def getEngineType(): String = {
+    Engine.getEngineType().toString
+  }
+
   def getNodeAndCoreNumber(): Array[Int] = {
     Array(Engine.nodeNumber(), Engine.coreNumber())
   }