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: 323906299
Change-Id: I92cb913fee27809dd3693ea3a47f45462a7e4794
This commit is contained in:
A. Unique TensorFlower 2020-07-29 17:47:04 -07:00 committed by TensorFlower Gardener
parent 14ea32321e
commit bb8d761d2e
1 changed files with 31 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -18416,6 +18416,37 @@ func SparseSegmentSum(scope *Scope, data tf.Output, indices tf.Output, segment_i
return op.Output(0)
}
// CollectiveReduceV2Attr is an optional argument to CollectiveReduceV2.
type CollectiveReduceV2Attr func(optionalAttr)
// CollectiveReduceV2CommunicationHint sets the optional communication_hint attribute to value.
// If not specified, defaults to "auto"
func CollectiveReduceV2CommunicationHint(value string) CollectiveReduceV2Attr {
return func(m optionalAttr) {
m["communication_hint"] = 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 {
return
}
attrs := map[string]interface{}{"merge_op": merge_op, "final_op": final_op}
for _, a := range optional {
a(attrs)
}
opspec := tf.OpSpec{
Type: "CollectiveReduceV2",
Input: []tf.Input{
input, group_size, group_key, instance_key,
},
Attrs: attrs,
}
op := scope.AddOperation(opspec)
return op.Output(0)
}
// Computes the sum along segments of a tensor.
//
// Read