experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Go: Update generated wrapper functions for TensorFlow ops.

Browse Source 
PiperOrigin-RevId: 234223332
This commit is contained in:
A. Unique TensorFlower 2019-02-15 15:22:52 -08:00 committed by TensorFlower Gardener
parent 6de3cb0023
commit 579fb17325
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/go/op/wrappers.go
View File

@ -19274,6 +19274,14 @@ func TPUReplicateMetadataPaddingMap(value []string) TPUReplicateMetadataAttr {
}
}
// TPUReplicateMetadataStepMarkerLocation sets the optional step_marker_location attribute to value.
// If not specified, defaults to "STEP_MARK_AT_ENTRY"
func TPUReplicateMetadataStepMarkerLocation(value string) TPUReplicateMetadataAttr {
return func(m optionalAttr) {
m["step_marker_location"] = value
}
}
// Metadata indicaitng how the TPU computation should be replicated.
//
// Arguments: