From 9cc1ff6768c14d2d0a01267c9c792055d382234f Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 11 Apr 2019 12:43:00 -0700 Subject: [PATCH] Go: Update generated wrapper functions for TensorFlow ops. PiperOrigin-RevId: 243122003 --- tensorflow/go/op/wrappers.go | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go index 99eceeb3980..db7aca695f0 100644 --- a/tensorflow/go/op/wrappers.go +++ b/tensorflow/go/op/wrappers.go @@ -30094,6 +30094,14 @@ func EnqueueTPUEmbeddingSparseTensorBatchCombiners(value []string) EnqueueTPUEmb } } +// EnqueueTPUEmbeddingSparseTensorBatchMaxSequenceLengths sets the optional max_sequence_lengths attribute to value. +// If not specified, defaults to <> +func EnqueueTPUEmbeddingSparseTensorBatchMaxSequenceLengths(value []int64) EnqueueTPUEmbeddingSparseTensorBatchAttr { + return func(m optionalAttr) { + m["max_sequence_lengths"] = value + } +} + // Eases the porting of code that uses tf.nn.embedding_lookup_sparse(). // // sample_indices[i], embedding_indices[i] and aggregation_weights[i] correspond @@ -38022,6 +38030,18 @@ func CombinedNonMaxSuppressionPadPerClass(value bool) CombinedNonMaxSuppressionA } } +// CombinedNonMaxSuppressionClipBoxes sets the optional clip_boxes attribute to value. +// +// value: If true, assume the box coordinates are between [0, 1] and clip the output boxes +// if they fall beyond [0, 1]. If false, do not do clipping and output the box +// coordinates as it is. +// If not specified, defaults to true +func CombinedNonMaxSuppressionClipBoxes(value bool) CombinedNonMaxSuppressionAttr { + return func(m optionalAttr) { + m["clip_boxes"] = value + } +} + // Greedily selects a subset of bounding boxes in descending order of score, // // This operation performs non_max_suppression on the inputs per batch, across