Add machine-readable list of PyTorch GPU tests.

Used internally for benchmarking.

PiperOrigin-RevId: 709153019

test/kubernetes/benchmarks/pytorch.go

@@ -144,6 +144,15 @@ var (
 			test:   eval,
 		},
 	}
+
+	// AllTests is a map of test names to the tests.
+	AllTests = map[string][]pytorchTest{
+		"FastNLPBert":       FastNLPBert,
+		"BigBird":           BigBird,
+		"SpeechTransformer": SpeechTransformer,
+		"LearningToPaint":   LearningToPaint,
+		"MobileNetV2":       MobileNetV2,
+	}
 )
 
 // Name returns the name of the test with the argument parameters included. It is formatted so