From 98e48ecca3cfd906c23756b4b4191da1b0784dd0 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Fri, 22 Jan 2021 20:46:19 -0800
Subject: [PATCH] Go: Update generated wrapper functions for TensorFlow ops.

PiperOrigin-RevId: 353368288
Change-Id: I9a24bce0df8b2011adb7857deb9dbf8306e60230
---
 tensorflow/go/op/wrappers.go | 18 +-----------------
 1 file changed, 1 insertion(+), 17 deletions(-)

diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go
index 64c61c1559d..b53e75f3af0 100644
--- a/tensorflow/go/op/wrappers.go
+++ b/tensorflow/go/op/wrappers.go
@@ -33195,32 +33195,16 @@ func BoostedTreesQuantileStreamResourceHandleOp(scope *Scope, optional ...Booste
 	return op.Output(0)
 }
 
-// XlaShardingAttr is an optional argument to XlaSharding.
-type XlaShardingAttr func(optionalAttr)
-
-// XlaShardingSharding sets the optional sharding attribute to value.
-// If not specified, defaults to ""
-func XlaShardingSharding(value string) XlaShardingAttr {
-	return func(m optionalAttr) {
-		m["sharding"] = value
-	}
-}
-
 // An op which shards the input based on the given sharding attribute.
-func XlaSharding(scope *Scope, input tf.Output, optional ...XlaShardingAttr) (output tf.Output) {
+func XlaSharding(scope *Scope, input tf.Output) (output tf.Output) {
 	if scope.Err() != nil {
 		return
 	}
-	attrs := map[string]interface{}{}
-	for _, a := range optional {
-		a(attrs)
-	}
 	opspec := tf.OpSpec{
 		Type: "XlaSharding",
 		Input: []tf.Input{
 			input,
 		},
-		Attrs: attrs,
 	}
 	op := scope.AddOperation(opspec)
 	return op.Output(0)