Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 243122003
This commit is contained in:
parent
fce41a2320
commit
9cc1ff6768
@ -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().
|
// Eases the porting of code that uses tf.nn.embedding_lookup_sparse().
|
||||||
//
|
//
|
||||||
// sample_indices[i], embedding_indices[i] and aggregation_weights[i] correspond
|
// 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,
|
// Greedily selects a subset of bounding boxes in descending order of score,
|
||||||
//
|
//
|
||||||
// This operation performs non_max_suppression on the inputs per batch, across
|
// This operation performs non_max_suppression on the inputs per batch, across
|
||||||
|
Loading…
Reference in New Issue
Block a user