tensorflow

Documentation

Index

• Variables
• func New(kubeClient framework.RuntimeClientInterface) framework.JobInterface
• type KubeTFJob
  • func (pj *KubeTFJob) AddEventListener(ctx context.Context, listenerType string, ...) error
  • func (pj *KubeTFJob) JobStatus(obj interface{}) (api.StatusInfo, error)
  • func (pj *KubeTFJob) Submit(ctx context.Context, job *api.PFJob) error

Variables

var (
	JobGVK          = k8s.TFJobGVK
	KubeTFFwVersion = client.KubeFrameworkVersion(JobGVK)
)

Functions

func New

func New(kubeClient framework.RuntimeClientInterface) framework.JobInterface

Types

type KubeTFJob

type KubeTFJob struct {
	kuberuntime.KubeBaseJob
}

KubeTFJob is a struct that runs a tensorflow job

func (*KubeTFJob) AddEventListener

func (pj *KubeTFJob) AddEventListener(ctx context.Context, listenerType string, jobQueue workqueue.RateLimitingInterface, listener interface{}) error

func (*KubeTFJob) JobStatus

func (pj *KubeTFJob) JobStatus(obj interface{}) (api.StatusInfo, error)

JobStatus get the statusInfo of TF job, including origin status, pf status and message

func (*KubeTFJob) Submit

func (pj *KubeTFJob) Submit(ctx context.Context, job *api.PFJob) error