Searched refs:TPUReplicateMetadataAttr (Results 1 – 1 of 1) sorted by relevance
34860 type TPUReplicateMetadataAttr func(optionalAttr) type34866 func TPUReplicateMetadataNumCoresPerReplica(value int64) TPUReplicateMetadataAttr {34876 func TPUReplicateMetadataTopology(value string) TPUReplicateMetadataAttr {34886 func TPUReplicateMetadataUseTpu(value bool) TPUReplicateMetadataAttr {34896 func TPUReplicateMetadataDeviceAssignment(value []int64) TPUReplicateMetadataAttr {34906 func TPUReplicateMetadataComputationShape(value []int64) TPUReplicateMetadataAttr {34914 func TPUReplicateMetadataHostComputeCore(value []string) TPUReplicateMetadataAttr {34922 func TPUReplicateMetadataPaddingMap(value []string) TPUReplicateMetadataAttr {34930 func TPUReplicateMetadataStepMarkerLocation(value string) TPUReplicateMetadataAttr {34938 func TPUReplicateMetadataAllowSoftPlacement(value bool) TPUReplicateMetadataAttr {[all …]