Add a new field SchedulerName to the controller config file format

This commit adds a new field SchedulerName to the controller config
file format. The purpose of the field is specifying the scheduler name
of the pods created by tf-operator and let the scheduler (which
wouldn't be the default scheduler) handle them. It would be convenient
for letting kube-batchd (a component of kube-arbitrator) handle the
pods.

pkg/apis/tensorflow/v1alpha1/types.go

@@ -184,6 +184,10 @@ type ControllerConfig struct {
 
 	// Path to the file containing the grpc server source
 	GrpcServerFilePath string
+
+	// SchedulerName will be assigned to v1.Pod.Spec.SchedulerName for specifying a scheduler
+	// which handles the pods of tf-operator
+	SchedulerName string
 }
 
 // AcceleratorVolume represents a host path that must be mounted into

pkg/trainer/replicas.go

@@ -177,6 +177,8 @@ func (s *TFReplicaSet) Create(config *tfv1alpha1.ControllerConfig) error {
 		// Make a copy of the template because we will modify it below. .
 		newPodSpecTemplate := s.Spec.Template.DeepCopy()
 
+		newPodSpecTemplate.Spec.SchedulerName = config.SchedulerName
+
 		newJ := &batch.Job{
 			ObjectMeta: meta_v1.ObjectMeta{
 				Name:   s.jobName(index),