Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/go/op/
Dwrappers.go20112 type TPUReplicateMetadataAttr func(optionalAttr) type
20118 func TPUReplicateMetadataNumCoresPerReplica(value int64) TPUReplicateMetadataAttr {
20128 func TPUReplicateMetadataTopology(value string) TPUReplicateMetadataAttr {
20138 func TPUReplicateMetadataUseTpu(value bool) TPUReplicateMetadataAttr {
20148 func TPUReplicateMetadataDeviceAssignment(value []int64) TPUReplicateMetadataAttr {
20158 func TPUReplicateMetadataComputationShape(value []int64) TPUReplicateMetadataAttr {
20166 func TPUReplicateMetadataHostComputeCore(value []string) TPUReplicateMetadataAttr {
20174 func TPUReplicateMetadataPaddingMap(value []string) TPUReplicateMetadataAttr {
20182 func TPUReplicateMetadataStepMarkerLocation(value string) TPUReplicateMetadataAttr {
20194 func TPUReplicateMetadata(scope *Scope, num_replicas int64, optional ...TPUReplicateMetadataAttr) (… argument