Skip to content

Commit faacdef

Browse files
committed
Fixed a bug where I was accidentally setting the shared name of input iterators in the learn API.
1 parent 4bb1f8b commit faacdef

File tree

1 file changed

+1
-1
lines changed
  • api/src/main/scala/org/platanios/tensorflow/api/learn/layers

1 file changed

+1
-1
lines changed

api/src/main/scala/org/platanios/tensorflow/api/learn/layers/Input.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ case class Input[T, O, DA, D, S](private val _dataType: DA, shape: S, name: Stri
4545

4646
private[this] val cache: mutable.Map[Graph, Iterator[T, O, D, S]] = mutable.Map.empty
4747

48-
protected def create(): Iterator[T, O, D, S] = Iterator.fromStructure(dataType, shape, name)
48+
protected def create(): Iterator[T, O, D, S] = Iterator.fromStructure(dataType, shape, name = name)
4949

5050
final def apply(): Iterator[T, O, D, S] = cache.getOrElse(Op.currentGraph, create())
5151

0 commit comments

Comments
 (0)