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: 354345029
Change-Id: Ie16dbe95b80d2e35056d12628b844a53bd8bed97
This commit is contained in:
A. Unique TensorFlower 2021-01-28 10:47:26 -08:00 committed by TensorFlower Gardener
parent f0442cc45f
commit d07f7fb70b
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -49035,6 +49035,14 @@ func EnqueueTPUEmbeddingSparseTensorBatchMaxSequenceLengths(value []int64) Enque
}
}
// EnqueueTPUEmbeddingSparseTensorBatchNumFeatures sets the optional num_features attribute to value.
// If not specified, defaults to <>
func EnqueueTPUEmbeddingSparseTensorBatchNumFeatures(value []int64) EnqueueTPUEmbeddingSparseTensorBatchAttr {
return func(m optionalAttr) {
m["num_features"] = value
}
}
// Eases the porting of code that uses tf.nn.embedding_lookup_sparse().
//
// sample_indices[i], embedding_indices[i] and aggregation_weights[i] correspond
@ -52301,6 +52309,14 @@ func EnqueueTPUEmbeddingRaggedTensorBatchMaxSequenceLengths(value []int64) Enque
}
}
// EnqueueTPUEmbeddingRaggedTensorBatchNumFeatures sets the optional num_features attribute to value.
// If not specified, defaults to <>
func EnqueueTPUEmbeddingRaggedTensorBatchNumFeatures(value []int64) EnqueueTPUEmbeddingRaggedTensorBatchAttr {
return func(m optionalAttr) {
m["num_features"] = value
}
}
// Eases the porting of code that uses tf.nn.embedding_lookup().
//
// sample_splits[i], embedding_indices[i] and aggregation_weights[i] correspond