bigdl keras private (intel#4731)

Le-Zheng · Sep 15, 2021 · ef4e98e · ef4e98e
1 parent 1a18e11
commit ef4e98e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/keras/Topology.scala b/scala/dllib/src/main/scala/com/intel/analytics/bigdl/dllib/keras/Topology.scala
@@ -166,7 +166,7 @@ abstract class KerasModel[T: ClassTag](implicit ev: TensorNumeric[T])
 @deprecated("`Model` is deprecated." +
   "com.intel.analytics.bigdl.dllib.keras is deprecated in BigDL 0.11, " +
   "and will be removed in future releases", "0.10.0")
-class Model[T: ClassTag](private val _inputs : Seq[ModuleNode[T]],
+private[bigdl] class Model[T: ClassTag](private val _inputs : Seq[ModuleNode[T]],
       private val _outputs : Seq[ModuleNode[T]])(implicit ev: TensorNumeric[T])
   extends KerasModel[T] {
   this.labor = doBuild(null)
@@ -266,7 +266,7 @@ object Model extends KerasLayerSerializable{
 @deprecated("`Sequential` is deprecated." +
   "com.intel.analytics.bigdl.dllib.keras is deprecated in BigDL 0.11, " +
   "and will be removed in future releases", "0.10.0")
-class Sequential[T: ClassTag]()
+private[bigdl] class Sequential[T: ClassTag]()
 (implicit ev: TensorNumeric[T]) extends KerasModel[T] {
 
   private[bigdl] var frozen: Boolean = false