From 5d3ef57b501781f1bab3b4ca85e8b8fc91671c14 Mon Sep 17 00:00:00 2001 From: Dan Zheng Date: Thu, 13 Jun 2019 13:14:09 -0700 Subject: [PATCH] Change `Differentiable.moved(along:)` to `move(along:)`. --- Sources/TensorFlow/Layers/Recurrent.swift | 2 +- Sources/TensorFlow/Optimizer.swift | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Sources/TensorFlow/Layers/Recurrent.swift b/Sources/TensorFlow/Layers/Recurrent.swift index 2fd18e9e9..fe802a228 100644 --- a/Sources/TensorFlow/Layers/Recurrent.swift +++ b/Sources/TensorFlow/Layers/Recurrent.swift @@ -88,7 +88,7 @@ public struct SimpleRNNCell: RNNCell, VectorPro // TODO(TF-507): Revert to `typealias State = Tensor` after // SR-10697 is fixed. public struct State: Equatable, Differentiable, VectorProtocol, KeyPathIterable { - public let value: Tensor + public var value: Tensor public init(_ value: Tensor) { self.value = value } diff --git a/Sources/TensorFlow/Optimizer.swift b/Sources/TensorFlow/Optimizer.swift index b58175f94..e59818a02 100644 --- a/Sources/TensorFlow/Optimizer.swift +++ b/Sources/TensorFlow/Optimizer.swift @@ -291,7 +291,7 @@ public class RiemannSGD: Optimizer public func update(_ model: inout Model.AllDifferentiableVariables, along direction: Model.TangentVector) { - model = model.moved(along: learningRate * (.zero - direction)) + model.move(along: learningRate * (.zero - direction)) } }