Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 224565479
This commit is contained in:
A. Unique TensorFlower
TensorFlower Gardener
parent
abc2610c5c
commit
32b1608308
@ -5862,6 +5862,121 @@ func MapStage(scope *Scope, key tf.Output, indices tf.Output, values []tf.Output
|
||||
return scope.AddOperation(opspec)
|
||||
}
|
||||
|
||||
// StageClearAttr is an optional argument to StageClear.
|
||||
type StageClearAttr func(optionalAttr)
|
||||
|
||||
// StageClearCapacity sets the optional capacity attribute to value.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageClearCapacity(value int64) StageClearAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["capacity"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageClearMemoryLimit sets the optional memory_limit attribute to value.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageClearMemoryLimit(value int64) StageClearAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["memory_limit"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageClearContainer sets the optional container attribute to value.
|
||||
// If not specified, defaults to ""
|
||||
func StageClearContainer(value string) StageClearAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["container"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageClearSharedName sets the optional shared_name attribute to value.
|
||||
// If not specified, defaults to ""
|
||||
func StageClearSharedName(value string) StageClearAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["shared_name"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// Op removes all elements in the underlying container.
|
||||
//
|
||||
// Returns the created operation.
|
||||
func StageClear(scope *Scope, dtypes []tf.DataType, optional ...StageClearAttr) (o *tf.Operation) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
attrs := map[string]interface{}{"dtypes": dtypes}
|
||||
for _, a := range optional {
|
||||
a(attrs)
|
||||
}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "StageClear",
|
||||
|
||||
Attrs: attrs,
|
||||
}
|
||||
return scope.AddOperation(opspec)
|
||||
}
|
||||
|
||||
// StageSizeAttr is an optional argument to StageSize.
|
||||
type StageSizeAttr func(optionalAttr)
|
||||
|
||||
// StageSizeCapacity sets the optional capacity attribute to value.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageSizeCapacity(value int64) StageSizeAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["capacity"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageSizeMemoryLimit sets the optional memory_limit attribute to value.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageSizeMemoryLimit(value int64) StageSizeAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["memory_limit"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageSizeContainer sets the optional container attribute to value.
|
||||
// If not specified, defaults to ""
|
||||
func StageSizeContainer(value string) StageSizeAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["container"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageSizeSharedName sets the optional shared_name attribute to value.
|
||||
// If not specified, defaults to ""
|
||||
func StageSizeSharedName(value string) StageSizeAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["shared_name"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// Op returns the number of elements in the underlying container.
|
||||
func StageSize(scope *Scope, dtypes []tf.DataType, optional ...StageSizeAttr) (size tf.Output) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
attrs := map[string]interface{}{"dtypes": dtypes}
|
||||
for _, a := range optional {
|
||||
a(attrs)
|
||||
}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "StageSize",
|
||||
|
||||
Attrs: attrs,
|
||||
}
|
||||
op := scope.AddOperation(opspec)
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// Compute the regularized incomplete beta integral \\(I_x(a, b)\\).
|
||||
//
|
||||
// The regularized incomplete beta integral is defined as:
|
||||
@ -11434,24 +11549,6 @@ func RandomShuffle(scope *Scope, value tf.Output, optional ...RandomShuffleAttr)
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// Returns the truth value of (x > y) element-wise.
|
||||
//
|
||||
// *NOTE*: `Greater` supports broadcasting. More about broadcasting
|
||||
// [here](http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html)
|
||||
func Greater(scope *Scope, x tf.Output, y tf.Output) (z tf.Output) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "Greater",
|
||||
Input: []tf.Input{
|
||||
x, y,
|
||||
},
|
||||
}
|
||||
op := scope.AddOperation(opspec)
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// ResourceSparseApplyRMSPropAttr is an optional argument to ResourceSparseApplyRMSProp.
|
||||
type ResourceSparseApplyRMSPropAttr func(optionalAttr)
|
||||
|
||||
@ -17210,64 +17307,6 @@ func NonMaxSuppressionWithOverlaps(scope *Scope, overlaps tf.Output, scores tf.O
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// StageClearAttr is an optional argument to StageClear.
|
||||
type StageClearAttr func(optionalAttr)
|
||||
|
||||
// StageClearCapacity sets the optional capacity attribute to value.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageClearCapacity(value int64) StageClearAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["capacity"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageClearMemoryLimit sets the optional memory_limit attribute to value.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageClearMemoryLimit(value int64) StageClearAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["memory_limit"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageClearContainer sets the optional container attribute to value.
|
||||
// If not specified, defaults to ""
|
||||
func StageClearContainer(value string) StageClearAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["container"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageClearSharedName sets the optional shared_name attribute to value.
|
||||
// If not specified, defaults to ""
|
||||
func StageClearSharedName(value string) StageClearAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["shared_name"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// Op removes all elements in the underlying container.
|
||||
//
|
||||
// Returns the created operation.
|
||||
func StageClear(scope *Scope, dtypes []tf.DataType, optional ...StageClearAttr) (o *tf.Operation) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
attrs := map[string]interface{}{"dtypes": dtypes}
|
||||
for _, a := range optional {
|
||||
a(attrs)
|
||||
}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "StageClear",
|
||||
|
||||
Attrs: attrs,
|
||||
}
|
||||
return scope.AddOperation(opspec)
|
||||
}
|
||||
|
||||
// Computes softmax cross entropy cost and gradients to backpropagate.
|
||||
//
|
||||
// Unlike `SoftmaxCrossEntropyWithLogits`, this operation does not accept
|
||||
@ -20236,63 +20275,6 @@ func NotEqual(scope *Scope, x tf.Output, y tf.Output) (z tf.Output) {
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// StageSizeAttr is an optional argument to StageSize.
|
||||
type StageSizeAttr func(optionalAttr)
|
||||
|
||||
// StageSizeCapacity sets the optional capacity attribute to value.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageSizeCapacity(value int64) StageSizeAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["capacity"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageSizeMemoryLimit sets the optional memory_limit attribute to value.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageSizeMemoryLimit(value int64) StageSizeAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["memory_limit"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageSizeContainer sets the optional container attribute to value.
|
||||
// If not specified, defaults to ""
|
||||
func StageSizeContainer(value string) StageSizeAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["container"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageSizeSharedName sets the optional shared_name attribute to value.
|
||||
// If not specified, defaults to ""
|
||||
func StageSizeSharedName(value string) StageSizeAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["shared_name"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// Op returns the number of elements in the underlying container.
|
||||
func StageSize(scope *Scope, dtypes []tf.DataType, optional ...StageSizeAttr) (size tf.Output) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
attrs := map[string]interface{}{"dtypes": dtypes}
|
||||
for _, a := range optional {
|
||||
a(attrs)
|
||||
}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "StageSize",
|
||||
|
||||
Attrs: attrs,
|
||||
}
|
||||
op := scope.AddOperation(opspec)
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// Produces the max pool of the input tensor for quantized types.
|
||||
//
|
||||
// Arguments:
|
||||
@ -31550,6 +31532,24 @@ func ModelDataset(scope *Scope, input_dataset tf.Output, output_types []tf.DataT
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// Returns the truth value of (x > y) element-wise.
|
||||
//
|
||||
// *NOTE*: `Greater` supports broadcasting. More about broadcasting
|
||||
// [here](http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html)
|
||||
func Greater(scope *Scope, x tf.Output, y tf.Output) (z tf.Output) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "Greater",
|
||||
Input: []tf.Input{
|
||||
x, y,
|
||||
},
|
||||
}
|
||||
op := scope.AddOperation(opspec)
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// Performs a padding as a preprocess during a convolution.
|
||||
//
|
||||
// Similar to FusedResizeAndPadConv2d, this op allows for an optimized
|
||||
|
||||
Loading…
Reference in New Issue
Block a user