Home
last modified time | relevance | path

Searched refs:TPUReplicateMetadataAttr (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/go/op/
Dwrappers.go34860 type TPUReplicateMetadataAttr func(optionalAttr) type
34866 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 …]