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) Delete(ctx context.Context, job *api.PFJob) error
  • func (pj *KubeTFJob) GetLog(ctx context.Context, jobLogRequest pfschema.JobLogRequest) (pfschema.JobLogInfo, error)
  • func (pj *KubeTFJob) JobStatus(obj interface{}) (api.StatusInfo, error)
  • func (pj *KubeTFJob) Stop(ctx context.Context, job *api.PFJob) error
  • func (pj *KubeTFJob) String(name string) string
  • func (pj *KubeTFJob) Submit(ctx context.Context, job *api.PFJob) error
  • func (pj *KubeTFJob) Update(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 {
	GVK schema.GroupVersionKind
	// contains filtered or unexported fields
}

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) Delete

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

func (*KubeTFJob) GetLog

func (pj *KubeTFJob) GetLog(ctx context.Context, jobLogRequest pfschema.JobLogRequest) (pfschema.JobLogInfo, 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) Stop

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

func (*KubeTFJob) String

func (pj *KubeTFJob) String(name string) string

func (*KubeTFJob) Submit

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

func (*KubeTFJob) Update

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