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: 327522072
Change-Id: I0647892a1a0fc3c78eb8625cbd534dada5dcde2c
This commit is contained in:
A. Unique TensorFlower 2020-08-19 15:46:35 -07:00 committed by TensorFlower Gardener
parent 78524e6f92
commit 345a0e60d0
1 changed files with 23 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15326,27 +15326,6 @@ func TensorMapHasKey(scope *Scope, input_handle tf.Output, key tf.Output) (has_k
return op.Output(0)
}
// Returns the value from a given key in a tensor map.
//
// input_handle: the input map
// key: the key to be looked up
// value: the value found from the given key
func TensorMapLookup(scope *Scope, input_handle tf.Output, key tf.Output, value_dtype tf.DataType) (value tf.Output) {
if scope.Err() != nil {
return
}
attrs := map[string]interface{}{"value_dtype": value_dtype}
opspec := tf.OpSpec{
Type: "TensorMapLookup",
Input: []tf.Input{
input_handle, key,
},
Attrs: attrs,
}
op := scope.AddOperation(opspec)
return op.Output(0)
}
// Inverse 3D fast Fourier transform.
//
// Computes the inverse 3-dimensional discrete Fourier transform over the
@ -15395,6 +15374,27 @@ func TensorMapInsert(scope *Scope, input_handle tf.Output, key tf.Output, value
return op.Output(0)
}
// Returns the value from a given key in a tensor map.
//
// input_handle: the input map
// key: the key to be looked up
// value: the value found from the given key
func TensorMapLookup(scope *Scope, input_handle tf.Output, key tf.Output, value_dtype tf.DataType) (value tf.Output) {
if scope.Err() != nil {
return
}
attrs := map[string]interface{}{"value_dtype": value_dtype}
opspec := tf.OpSpec{
Type: "TensorMapLookup",
Input: []tf.Input{
input_handle, key,
},
Attrs: attrs,
}
op := scope.AddOperation(opspec)
return op.Output(0)
}
// Merges summaries.
//
// This op creates a
@ -20366,8 +20366,7 @@ func Polygamma(scope *Scope, a tf.Output, x tf.Output) (z tf.Output) {
// input_handle: the original map
// output_handle: the map with value from given key removed
// key: the key of the value to be erased
// value: the value that was erased
func TensorMapErase(scope *Scope, input_handle tf.Output, key tf.Output, value_dtype tf.DataType) (output_handle tf.Output, value tf.Output) {
func TensorMapErase(scope *Scope, input_handle tf.Output, key tf.Output, value_dtype tf.DataType) (output_handle tf.Output) {
if scope.Err() != nil {
return
}
@ -20380,7 +20379,7 @@ func TensorMapErase(scope *Scope, input_handle tf.Output, key tf.Output, value_d
Attrs: attrs,
}
op := scope.AddOperation(opspec)
return op.Output(0), op.Output(1)
return op.Output(0)
}
// Shuffle dimensions of x according to a permutation.