blob: e28cbeac9d2846ee4ee690a93b8a1cc7ff503877 [file] [log] [blame]
struct TF_619: Differentiable {
var p: Float = 1
@differentiable
func foo(_ x: Float) -> Float {
return p * x
}
@derivative(of: foo)
func vjpFoo(_ x: Float) -> (value: Float, pullback: (Float) -> (TangentVector, Float)) {
return (x, { v in (TangentVector(p: v * x), v * self.p) })
}
}