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: 224565479
This commit is contained in:
A. Unique TensorFlower 2018-12-07 13:11:24 -08:00 committed by TensorFlower Gardener
parent abc2610c5c
commit 32b1608308
1 changed files with 133 additions and 133 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -5862,6 +5862,121 @@ func MapStage(scope *Scope, key tf.Output, indices tf.Output, values []tf.Output
return scope.AddOperation(opspec) 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)\\). // Compute the regularized incomplete beta integral \\(I_x(a, b)\\).
// //
// The regularized incomplete beta integral is defined as: // 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) 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. // ResourceSparseApplyRMSPropAttr is an optional argument to ResourceSparseApplyRMSProp.
type ResourceSparseApplyRMSPropAttr func(optionalAttr) type ResourceSparseApplyRMSPropAttr func(optionalAttr)
@ -17210,64 +17307,6 @@ func NonMaxSuppressionWithOverlaps(scope *Scope, overlaps tf.Output, scores tf.O
return op.Output(0) 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. // Computes softmax cross entropy cost and gradients to backpropagate.
// //
// Unlike `SoftmaxCrossEntropyWithLogits`, this operation does not accept // 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) 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. // Produces the max pool of the input tensor for quantized types.
// //
// Arguments: // Arguments:
@ -31550,6 +31532,24 @@ func ModelDataset(scope *Scope, input_dataset tf.Output, output_types []tf.DataT
return op.Output(0) 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. // Performs a padding as a preprocess during a convolution.
// //
// Similar to FusedResizeAndPadConv2d, this op allows for an optimized // Similar to FusedResizeAndPadConv2d, this op allows for an optimized