Go: Update generated wrapper functions for TensorFlow ops.
PiperOrigin-RevId: 247050991
This commit is contained in:
A. Unique TensorFlower
TensorFlower Gardener
parent
60ac1b8adf
commit
098ff16ec9
@ -8693,101 +8693,6 @@ func WindowDataset(scope *Scope, input_dataset tf.Output, size tf.Output, shift
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// StageAttr is an optional argument to Stage.
|
||||
type StageAttr func(optionalAttr)
|
||||
|
||||
// StageCapacity sets the optional capacity attribute to value.
|
||||
//
|
||||
// value: Maximum number of elements in the Staging Area. If > 0, inserts
|
||||
// on the container will block when the capacity is reached.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageCapacity(value int64) StageAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["capacity"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageMemoryLimit sets the optional memory_limit attribute to value.
|
||||
//
|
||||
// value: The maximum number of bytes allowed for Tensors in the Staging Area.
|
||||
// If > 0, inserts will block until sufficient space is available.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageMemoryLimit(value int64) StageAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["memory_limit"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageContainer sets the optional container attribute to value.
|
||||
//
|
||||
// value: If non-empty, this queue is placed in the given container. Otherwise,
|
||||
// a default container is used.
|
||||
// If not specified, defaults to ""
|
||||
func StageContainer(value string) StageAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["container"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageSharedName sets the optional shared_name attribute to value.
|
||||
//
|
||||
// value: It is necessary to match this name to the matching Unstage Op.
|
||||
// If not specified, defaults to ""
|
||||
func StageSharedName(value string) StageAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["shared_name"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// Stage values similar to a lightweight Enqueue.
|
||||
//
|
||||
// The basic functionality of this Op is similar to a queue with many
|
||||
// fewer capabilities and options. This Op is optimized for performance.
|
||||
//
|
||||
// Arguments:
|
||||
// values: a list of tensors
|
||||
// dtypes A list of data types that inserted values should adhere to.
|
||||
//
|
||||
// Returns the created operation.
|
||||
func Stage(scope *Scope, values []tf.Output, optional ...StageAttr) (o *tf.Operation) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
attrs := map[string]interface{}{}
|
||||
for _, a := range optional {
|
||||
a(attrs)
|
||||
}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "Stage",
|
||||
Input: []tf.Input{
|
||||
tf.OutputList(values),
|
||||
},
|
||||
Attrs: attrs,
|
||||
}
|
||||
return scope.AddOperation(opspec)
|
||||
}
|
||||
|
||||
// Creates a dataset containing elements of first component of `input_dataset` having true in the last component.
|
||||
func FilterByLastComponentDataset(scope *Scope, input_dataset tf.Output, output_types []tf.DataType, output_shapes []tf.Shape) (output tf.Output) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
attrs := map[string]interface{}{"output_types": output_types, "output_shapes": output_shapes}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "FilterByLastComponentDataset",
|
||||
Input: []tf.Input{
|
||||
input_dataset,
|
||||
},
|
||||
Attrs: attrs,
|
||||
}
|
||||
op := scope.AddOperation(opspec)
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// Produces the average pool of the input tensor for quantized types.
|
||||
//
|
||||
// Arguments:
|
||||
@ -40250,6 +40155,101 @@ func SamplingDataset(scope *Scope, input_dataset tf.Output, rate tf.Output, seed
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// StageAttr is an optional argument to Stage.
|
||||
type StageAttr func(optionalAttr)
|
||||
|
||||
// StageCapacity sets the optional capacity attribute to value.
|
||||
//
|
||||
// value: Maximum number of elements in the Staging Area. If > 0, inserts
|
||||
// on the container will block when the capacity is reached.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageCapacity(value int64) StageAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["capacity"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageMemoryLimit sets the optional memory_limit attribute to value.
|
||||
//
|
||||
// value: The maximum number of bytes allowed for Tensors in the Staging Area.
|
||||
// If > 0, inserts will block until sufficient space is available.
|
||||
// If not specified, defaults to 0
|
||||
//
|
||||
// REQUIRES: value >= 0
|
||||
func StageMemoryLimit(value int64) StageAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["memory_limit"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageContainer sets the optional container attribute to value.
|
||||
//
|
||||
// value: If non-empty, this queue is placed in the given container. Otherwise,
|
||||
// a default container is used.
|
||||
// If not specified, defaults to ""
|
||||
func StageContainer(value string) StageAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["container"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// StageSharedName sets the optional shared_name attribute to value.
|
||||
//
|
||||
// value: It is necessary to match this name to the matching Unstage Op.
|
||||
// If not specified, defaults to ""
|
||||
func StageSharedName(value string) StageAttr {
|
||||
return func(m optionalAttr) {
|
||||
m["shared_name"] = value
|
||||
}
|
||||
}
|
||||
|
||||
// Stage values similar to a lightweight Enqueue.
|
||||
//
|
||||
// The basic functionality of this Op is similar to a queue with many
|
||||
// fewer capabilities and options. This Op is optimized for performance.
|
||||
//
|
||||
// Arguments:
|
||||
// values: a list of tensors
|
||||
// dtypes A list of data types that inserted values should adhere to.
|
||||
//
|
||||
// Returns the created operation.
|
||||
func Stage(scope *Scope, values []tf.Output, optional ...StageAttr) (o *tf.Operation) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
attrs := map[string]interface{}{}
|
||||
for _, a := range optional {
|
||||
a(attrs)
|
||||
}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "Stage",
|
||||
Input: []tf.Input{
|
||||
tf.OutputList(values),
|
||||
},
|
||||
Attrs: attrs,
|
||||
}
|
||||
return scope.AddOperation(opspec)
|
||||
}
|
||||
|
||||
// Creates a dataset containing elements of first component of `input_dataset` having true in the last component.
|
||||
func FilterByLastComponentDataset(scope *Scope, input_dataset tf.Output, output_types []tf.DataType, output_shapes []tf.Shape) (output tf.Output) {
|
||||
if scope.Err() != nil {
|
||||
return
|
||||
}
|
||||
attrs := map[string]interface{}{"output_types": output_types, "output_shapes": output_shapes}
|
||||
opspec := tf.OpSpec{
|
||||
Type: "FilterByLastComponentDataset",
|
||||
Input: []tf.Input{
|
||||
input_dataset,
|
||||
},
|
||||
Attrs: attrs,
|
||||
}
|
||||
op := scope.AddOperation(opspec)
|
||||
return op.Output(0)
|
||||
}
|
||||
|
||||
// EncodeProtoAttr is an optional argument to EncodeProto.
|
||||
type EncodeProtoAttr func(optionalAttr)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user