diff --git a/tensorflow/go/op/wrappers.go b/tensorflow/go/op/wrappers.go index 98f62805864..2a4b4065464 100644 --- a/tensorflow/go/op/wrappers.go +++ b/tensorflow/go/op/wrappers.go @@ -13634,6 +13634,33 @@ func QueueDequeueV2(scope *Scope, handle tf.Output, component_types []tf.DataTyp return components } +// Returns the next record (key, value pair) produced by a Reader. +// +// Will dequeue from the input queue if necessary (e.g. when the +// Reader needs to start reading from a new file since it has finished +// with the previous file). +// +// Arguments: +// reader_handle: Handle to a Reader. +// queue_handle: Handle to a Queue, with string work items. +// +// Returns: +// key: A scalar. +// value: A scalar. +func ReaderReadV2(scope *Scope, reader_handle tf.Output, queue_handle tf.Output) (key tf.Output, value tf.Output) { + if scope.Err() != nil { + return + } + opspec := tf.OpSpec{ + Type: "ReaderReadV2", + Input: []tf.Input{ + reader_handle, queue_handle, + }, + } + op := scope.AddOperation(opspec) + return op.Output(0), op.Output(1) +} + // Return a slice from 'input'. // // The output tensor is a tensor with dimensions described by 'size' @@ -15927,6 +15954,46 @@ func Dilation2D(scope *Scope, input tf.Output, filter tf.Output, strides []int64 return op.Output(0) } +// IsotonicRegressionAttr is an optional argument to IsotonicRegression. +type IsotonicRegressionAttr func(optionalAttr) + +// IsotonicRegressionOutputDtype sets the optional output_dtype attribute to value. +// +// value: Dtype of output. +// If not specified, defaults to DT_FLOAT +func IsotonicRegressionOutputDtype(value tf.DataType) IsotonicRegressionAttr { + return func(m optionalAttr) { + m["output_dtype"] = value + } +} + +// Solves a batch of isotonic regression problems. +// +// Arguments: +// input: A (batch_size, dim)-tensor holding a batch of inputs. +// +// Returns: +// output: A (batch_size, dim)-tensor holding the per-batch element solutions. +// segments: An int32 (batch_size, dim)-tensor with the segments. +func IsotonicRegression(scope *Scope, input tf.Output, optional ...IsotonicRegressionAttr) (output tf.Output, segments tf.Output) { + if scope.Err() != nil { + return + } + attrs := map[string]interface{}{} + for _, a := range optional { + a(attrs) + } + opspec := tf.OpSpec{ + Type: "IsotonicRegression", + Input: []tf.Input{ + input, + }, + Attrs: attrs, + } + op := scope.AddOperation(opspec) + return op.Output(0), op.Output(1) +} + // Computes softplus: `log(exp(features) + 1)`. func Softplus(scope *Scope, features tf.Output) (activations tf.Output) { if scope.Err() != nil { @@ -49688,33 +49755,6 @@ func LoadTPUEmbeddingMDLAdagradLightParameters(scope *Scope, parameters tf.Outpu return scope.AddOperation(opspec) } -// Returns the next record (key, value pair) produced by a Reader. -// -// Will dequeue from the input queue if necessary (e.g. when the -// Reader needs to start reading from a new file since it has finished -// with the previous file). -// -// Arguments: -// reader_handle: Handle to a Reader. -// queue_handle: Handle to a Queue, with string work items. -// -// Returns: -// key: A scalar. -// value: A scalar. -func ReaderReadV2(scope *Scope, reader_handle tf.Output, queue_handle tf.Output) (key tf.Output, value tf.Output) { - if scope.Err() != nil { - return - } - opspec := tf.OpSpec{ - Type: "ReaderReadV2", - Input: []tf.Input{ - reader_handle, queue_handle, - }, - } - op := scope.AddOperation(opspec) - return op.Output(0), op.Output(1) -} - // CumprodAttr is an optional argument to Cumprod. type CumprodAttr func(optionalAttr)