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: 248624743
This commit is contained in:
A. Unique TensorFlower 2019-05-16 16:35:42 -07:00 committed by TensorFlower Gardener
parent 7bb6edf4a7
commit fb20a668e8
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -8206,6 +8206,23 @@ func WindowDataset(scope *Scope, input_dataset tf.Output, size tf.Output, shift
return op.Output(0)
}
// Creates a dataset containing elements of first component of `input_dataset` having true in the last component.
func FilterByLastComponentDataset(scope *Scope, input_dataset tf.Output, output_types []tf.DataType, output_shapes []tf.Shape) (output tf.Output) {
if scope.Err() != nil {
return
}
attrs := map[string]interface{}{"output_types": output_types, "output_shapes": output_shapes}
opspec := tf.OpSpec{
Type: "FilterByLastComponentDataset",
Input: []tf.Input{
input_dataset,
},
Attrs: attrs,
}
op := scope.AddOperation(opspec)
return op.Output(0)
}
// QuantizedRelu6Attr is an optional argument to QuantizedRelu6.
type QuantizedRelu6Attr func(optionalAttr)
@ -40491,20 +40508,3 @@ func Square(scope *Scope, x tf.Output) (y tf.Output) {
op := scope.AddOperation(opspec)
return op.Output(0)
}
// Creates a dataset containing elements of first component of `input_dataset` having true in the last component.
func FilterByLastComponentDataset(scope *Scope, input_dataset tf.Output, output_types []tf.DataType, output_shapes []tf.Shape) (output tf.Output) {
if scope.Err() != nil {
return
}
attrs := map[string]interface{}{"output_types": output_types, "output_shapes": output_shapes}
opspec := tf.OpSpec{
Type: "FilterByLastComponentDataset",
Input: []tf.Input{
input_dataset,
},
Attrs: attrs,
}
op := scope.AddOperation(opspec)
return op.Output(0)
}