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

Author: pschuh

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
@@ -177,7 +177,7 @@ func _vjpSplitQKV(_ input: Tensor<Float>)
     })
 }
 
-struct MultiHeadAttention: Layer {
+struct MultiHeadAttention: ParameterlessLayer {
     var attention: Attention
     var wqkv: TimeDistributed
     var wo: TimeDistributed
@@ -213,7 +213,7 @@ struct MultiHeadAttention: Layer {
     }
 }
 
-struct EncoderLayer: Layer {
+struct EncoderLayer: ParameterlessLayer {
     var selfAttention: MultiHeadAttention
     var selfAttentionDropout: Dropout<Float>
     var selfAttentionNorm: LayerNorm<Float>

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