nightlies URL: s4tf-kokoro-artifact-testing => swift-tensorflow-artifacts

pschuh · pschuh · commit 0de86fec3c0f · 2019-07-29T15:51:57.000-07:00
diff --git a/Transformer/Model.swift b/Transformer/Model.swift
@@ -105,7 +105,7 @@ func _vjpCausallyMasked(_ dotProducts: Tensor<Float>, enable: Bool)
     return (causallyMasked(dotProducts), identity)
 }
 
-struct Attention: Layer {
+struct Attention: ParameterlessLayer {
     @noDerivative let dropout: Dropout<Float>
     @noDerivative let scale: Tensor<Float>
     @noDerivative let causal: Bool
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -3,7 +3,7 @@
 FROM nvidia/cuda:9.2-cudnn7-devel-ubuntu18.04
 
 # Allows the caller to specify the toolchain to use.
-ARG swift_tf_url=https://storage.googleapis.com/s4tf-kokoro-artifact-testing/latest/swift-tensorflow-DEVELOPMENT-ubuntu18.04.tar.gz
+ARG swift_tf_url=https://storage.googleapis.com/swift-tensorflow-artifacts/nightlies/latest/swift-tensorflow-DEVELOPMENT-ubuntu18.04.tar.gz
 
 # Install Swift deps.
 ENV DEBIAN_FRONTEND=noninteractive

Original file line number	Diff line number	Diff line change
`@@ -105,7 +105,7 @@ func _vjpCausallyMasked(_ dotProducts: Tensor<Float>, enable: Bool)`
`105`	`105`	`return (causallyMasked(dotProducts), identity)`
`106`	`106`	`}`
`107`	`107`
`108`		`-struct Attention: Layer {`
	`108`	`+struct Attention: ParameterlessLayer {`
`109`	`109`	`@noDerivative let dropout: Dropout<Float>`
`110`	`110`	`@noDerivative let scale: Tensor<Float>`
`111`	`111`	`@noDerivative let causal: Bool`