Fix tensor devices for DARTS Trial #1061

Job	Run time
Publish Image (darts-cnn-cifar10-cpu, linux/amd64,linux/arm64, examples/v1beta1/trial-images/dart... / Publish Image	6m 35s
Publish Image (darts-cnn-cifar10-gpu, linux/amd64, examples/v1beta1/trial-images/darts-cnn-cifar1... / Publish Image	10m 36s
Publish Image (enas-cnn-cifar10-cpu, linux/amd64,linux/arm64, examples/v1beta1/trial-images/enas-... / Publish Image	10m 26s
Publish Image (enas-cnn-cifar10-gpu, linux/amd64, examples/v1beta1/trial-images/enas-cnn-cifar10/... / Publish Image	6m 49s
Publish Image (pytorch-mnist-cpu, linux/amd64,linux/arm64, examples/v1beta1/trial-images/pytorch-... / Publish Image	5m 56s
Publish Image (pytorch-mnist-gpu, linux/amd64, examples/v1beta1/trial-images/pytorch-mnist/Docker... / Publish Image	10m 8s
Publish Image (simple-pbt, linux/amd64,linux/arm64, examples/v1beta1/trial-images/simple-pbt/Dock... / Publish Image	1m 49s
Publish Image (tf-mnist-with-summaries, linux/amd64,linux/arm64, examples/v1beta1/trial-images/tf... / Publish Image	6m 5s
	58m 24s

Provide feedback