diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go
index ca718602d07..6ae011b81f0 100644
--- a/tensorflow/go/op/wrappers.go
+++ b/tensorflow/go/op/wrappers.go
@@ -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: