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: 239214941
This commit is contained in:
A. Unique TensorFlower 2019-03-19 10:10:28 -07:00 committed by TensorFlower Gardener
parent b55673b490
commit a988f5e6fa
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

@ -7652,6 +7652,17 @@ func OptionalFromValue(scope *Scope, components []tf.Output) (optional tf.Output
return op.Output(0) return op.Output(0)
} }
// OptimizeDatasetAttr is an optional argument to OptimizeDataset.
type OptimizeDatasetAttr func(optionalAttr)
// OptimizeDatasetOptimizationConfigs sets the optional optimization_configs attribute to value.
// If not specified, defaults to <>
func OptimizeDatasetOptimizationConfigs(value []string) OptimizeDatasetAttr {
return func(m optionalAttr) {
m["optimization_configs"] = value
}
}
// Creates a dataset by applying optimizations to `input_dataset`. // Creates a dataset by applying optimizations to `input_dataset`.
// //
// Creates a dataset by applying optimizations to `input_dataset`. // Creates a dataset by applying optimizations to `input_dataset`.
@ -7661,11 +7672,14 @@ func OptionalFromValue(scope *Scope, components []tf.Output) (optional tf.Output
// optimizations: A `tf.string` vector `tf.Tensor` identifying optimizations to use. // optimizations: A `tf.string` vector `tf.Tensor` identifying optimizations to use.
// //
// //
func OptimizeDataset(scope *Scope, input_dataset tf.Output, optimizations tf.Output, output_types []tf.DataType, output_shapes []tf.Shape) (handle tf.Output) { func OptimizeDataset(scope *Scope, input_dataset tf.Output, optimizations tf.Output, output_types []tf.DataType, output_shapes []tf.Shape, optional ...OptimizeDatasetAttr) (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: "OptimizeDataset", Type: "OptimizeDataset",
Input: []tf.Input{ Input: []tf.Input{