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: 245289270
This commit is contained in:
A. Unique TensorFlower 2019-04-25 12:49:37 -07:00 committed by TensorFlower Gardener
parent a74f9c3c61
commit 1ff68d121e
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -23501,6 +23501,17 @@ func Range(scope *Scope, start tf.Output, limit tf.Output, delta tf.Output) (out
return op.Output(0) return op.Output(0)
} }
// ShardDatasetAttr is an optional argument to ShardDataset.
type ShardDatasetAttr func(optionalAttr)
// ShardDatasetRequireNonEmpty sets the optional require_non_empty attribute to value.
// If not specified, defaults to false
func ShardDatasetRequireNonEmpty(value bool) ShardDatasetAttr {
return func(m optionalAttr) {
m["require_non_empty"] = value
}
}
// Creates a `Dataset` that includes only 1/`num_shards` of this dataset. // Creates a `Dataset` that includes only 1/`num_shards` of this dataset.
// //
// Arguments: // Arguments:
@ -23509,11 +23520,14 @@ func Range(scope *Scope, start tf.Output, limit tf.Output, delta tf.Output) (out
// index: An integer representing the current worker index. // index: An integer representing the current worker index.
// //
// //
func ShardDataset(scope *Scope, input_dataset tf.Output, num_shards tf.Output, index tf.Output, output_types []tf.DataType, output_shapes []tf.Shape) (handle tf.Output) { func ShardDataset(scope *Scope, input_dataset tf.Output, num_shards tf.Output, index tf.Output, output_types []tf.DataType, output_shapes []tf.Shape, optional ...ShardDatasetAttr) (handle tf.Output) {
if scope.Err() != nil { if scope.Err() != nil {
return return
} }
attrs := map[string]interface{}{"output_types": output_types, "output_shapes": output_shapes} attrs := map[string]interface{}{"output_types": output_types, "output_shapes": output_shapes}
for _, a := range optional {
a(attrs)
}
opspec := tf.OpSpec{ opspec := tf.OpSpec{
Type: "ShardDataset", Type: "ShardDataset",
Input: []tf.Input{ Input: []tf.Input{