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: 353368288
Change-Id: I9a24bce0df8b2011adb7857deb9dbf8306e60230
This commit is contained in:
A. Unique TensorFlower 2021-01-22 20:46:19 -08:00 committed by TensorFlower Gardener
parent d56de8e17e
commit 98e48ecca3
1 changed files with 1 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -33195,32 +33195,16 @@ func BoostedTreesQuantileStreamResourceHandleOp(scope *Scope, optional ...Booste
return op.Output(0) 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. // 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 { if scope.Err() != nil {
return return
} }
attrs := map[string]interface{}{}
for _, a := range optional {
a(attrs)
}
opspec := tf.OpSpec{ opspec := tf.OpSpec{
Type: "XlaSharding", Type: "XlaSharding",
Input: []tf.Input{ Input: []tf.Input{
input, input,
}, },
Attrs: attrs,
} }
op := scope.AddOperation(opspec) op := scope.AddOperation(opspec)
return op.Output(0) return op.Output(0)