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: 334453529
Change-Id: Ie8efc2d3fb2c6ced3b73cba7075f7323958f2a8b
This commit is contained in:
A. Unique TensorFlower 2020-09-29 13:46:02 -07:00 committed by TensorFlower Gardener
parent 81a44fd840
commit 9ca20379fc
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

@ -18807,6 +18807,14 @@ func CollectiveReduceV2CommunicationHint(value string) CollectiveReduceV2Attr {
}
}
// CollectiveReduceV2TimeoutSeconds sets the optional timeout_seconds attribute to value.
// If not specified, defaults to 0
func CollectiveReduceV2TimeoutSeconds(value float32) CollectiveReduceV2Attr {
return func(m optionalAttr) {
m["timeout_seconds"] = value
}
}
// Mutually reduces multiple tensors of identical type and shape.
func CollectiveReduceV2(scope *Scope, input tf.Output, group_size tf.Output, group_key tf.Output, instance_key tf.Output, merge_op string, final_op string, optional ...CollectiveReduceV2Attr) (data tf.Output) {
if scope.Err() != nil {