tensorflow

Documentation

Index

• Constants
• type TensorFlowGrouper
  • func NewTensorFlowGrouper(kubeflowGrouper *kubeflow.KubeflowDistributedGrouper) *TensorFlowGrouper
  • func (tfg *TensorFlowGrouper) GetPodGroupMetadata(topOwner *unstructured.Unstructured, pod *v1.Pod, ...) (*podgroup.Metadata, error)
  • func (tfg *TensorFlowGrouper) Name() string

Constants

const (
	ReplicaSpecName = "tfReplicaSpecs"
)

Types

type TensorFlowGrouper

type TensorFlowGrouper struct {
	*kubeflow.KubeflowDistributedGrouper
}

func NewTensorFlowGrouper

func NewTensorFlowGrouper(kubeflowGrouper *kubeflow.KubeflowDistributedGrouper) *TensorFlowGrouper

func (*TensorFlowGrouper) GetPodGroupMetadata

func (tfg *TensorFlowGrouper) GetPodGroupMetadata(
	topOwner *unstructured.Unstructured, pod *v1.Pod, _ ...*metav1.PartialObjectMetadata,
) (*podgroup.Metadata, error)

func (*TensorFlowGrouper) Name

func (tfg *TensorFlowGrouper) Name() string