feat: multi models support with MKL-DNN backend (intel-analytics#2936)

* feat: multi models support with MKL-DNN backend
dding3 · Oct 26, 2019 · 149ce12 · 149ce12
1 parent 63c092b
commit 149ce12
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/dl/src/main/scala/com/intel/analytics/bigdl/dllib/tensor/DnnTensor.scala b/dl/src/main/scala/com/intel/analytics/bigdl/dllib/tensor/DnnTensor.scala
@@ -162,6 +162,13 @@ class DnnTensor[T: ClassTag](
     this
   }
 
+  override def set(other: Tensor[T]): Tensor[T] = {
+    require(other.isInstanceOf[DnnTensor[T]], s"only support to set DnnTensor")
+    this._storage.release()
+    this._storage = other.storage().asInstanceOf[DnnStorage[T]]
+    this
+  }
+
   override def toString: String = {
     ev.getType() match {
       case FloatType =>