experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Go: Update generated wrapper functions for TensorFlow ops. 

PiperOrigin-RevId: 311478670
Change-Id: Ib8c15d5cba307629a0d8fc55e07efc401502899e
This commit is contained in:
A. Unique TensorFlower 2020-05-13 23:46:17 -07:00 committed by TensorFlower Gardener
parent 112288586d
commit 03f3e8153c
1 changed files with 65 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -4715,7 +4715,7 @@ type DenseCountSparseOutputAttr func(optionalAttr)
// DenseCountSparseOutputMinlength sets the optional minlength attribute to value. // DenseCountSparseOutputMinlength sets the optional minlength attribute to value.
// //
// value: Minimum value to count. Can be set to -1 for no minimum. // value: int32; minimum value to count. Can be set to -1 for no minimum.
// If not specified, defaults to -1 // If not specified, defaults to -1
// //
// REQUIRES: value >= -1 // REQUIRES: value >= -1
@ -4727,7 +4727,7 @@ func DenseCountSparseOutputMinlength(value int64) DenseCountSparseOutputAttr {
// DenseCountSparseOutputMaxlength sets the optional maxlength attribute to value. // DenseCountSparseOutputMaxlength sets the optional maxlength attribute to value.
// //
// value: Maximum value to count. Can be set to -1 for no maximum. // value: int32; maximum value to count. Can be set to -1 for no maximum.
// If not specified, defaults to -1 // If not specified, defaults to -1
// //
// REQUIRES: value >= -1 // REQUIRES: value >= -1
@ -4742,20 +4742,20 @@ func DenseCountSparseOutputMaxlength(value int64) DenseCountSparseOutputAttr {
// Counts the number of times each value occurs in the input. // Counts the number of times each value occurs in the input.
// //
// Arguments: // Arguments:
// values: Tensor containing data to count. // values: int32 or int64; Tensor containing data to count.
// weights: A Tensor of the same shape as indices containing per-index weight values. May // weights: float32; Optional rank 1 Tensor (shape=[max_values]) with weights for each count value.
// also be the empty tensor if no weights are used. // binary_count: bool; whether to output the number of occurrences of each value or 1.
// binary_output: Whether to output the number of occurrences of each value or 1. // output_type: dtype; dtype of the output values tensor.
// //
// Returns: // Returns:
// output_indices: Indices tensor for the resulting sparse tensor object. // output_indices: int64; indices tensor for the resulting sparse tensor object.
// output_values: Values tensor for the resulting sparse tensor object. // output_values: int64 or float32; values tensor for the resulting sparse tensor object.
// output_dense_shape: Shape tensor for the resulting sparse tensor object. // output_dense_shape: int64; shape tensor for the resulting sparse tensor object.
func DenseCountSparseOutput(scope *Scope, values tf.Output, weights tf.Output, binary_output bool, optional ...DenseCountSparseOutputAttr) (output_indices tf.Output, output_values tf.Output, output_dense_shape tf.Output) { func DenseCountSparseOutput(scope *Scope, values tf.Output, weights tf.Output, binary_count bool, output_type tf.DataType, optional ...DenseCountSparseOutputAttr) (output_indices tf.Output, output_values tf.Output, output_dense_shape tf.Output) {
if scope.Err() != nil { if scope.Err() != nil {
return return
} }
attrs := map[string]interface{}{"binary_output": binary_output} attrs := map[string]interface{}{"binary_count": binary_count, "output_type": output_type}
for _, a := range optional { for _, a := range optional {
a(attrs) a(attrs)
} }
@ -8607,7 +8607,7 @@ type RaggedCountSparseOutputAttr func(optionalAttr)
// RaggedCountSparseOutputMinlength sets the optional minlength attribute to value. // RaggedCountSparseOutputMinlength sets the optional minlength attribute to value.
// //
// value: Minimum value to count. Can be set to -1 for no minimum. // value: int32; minimum value to count. Can be set to -1 for no minimum.
// If not specified, defaults to -1 // If not specified, defaults to -1
// //
// REQUIRES: value >= -1 // REQUIRES: value >= -1
@ -8619,7 +8619,7 @@ func RaggedCountSparseOutputMinlength(value int64) RaggedCountSparseOutputAttr {
// RaggedCountSparseOutputMaxlength sets the optional maxlength attribute to value. // RaggedCountSparseOutputMaxlength sets the optional maxlength attribute to value.
// //
// value: Maximum value to count. Can be set to -1 for no maximum. // value: int32; maximum value to count. Can be set to -1 for no maximum.
// If not specified, defaults to -1 // If not specified, defaults to -1
// //
// REQUIRES: value >= -1 // REQUIRES: value >= -1
@ -8634,27 +8634,33 @@ func RaggedCountSparseOutputMaxlength(value int64) RaggedCountSparseOutputAttr {
// Counts the number of times each value occurs in the input. // Counts the number of times each value occurs in the input.
// //
// Arguments: // Arguments:
// splits: Tensor containing the row splits of the ragged tensor to count. // splits: int64; Tensor containing the row splits of the ragged tensor to count.
// values: Tensor containing values of the sparse tensor to count. // values: int32 or int64; Tensor containing values of the sparse tensor to count.
// weights: A Tensor of the same shape as indices containing per-index weight values. // weights: float32; Optional rank 1 Tensor (shape=[max_values]) with weights for each count value.
// May also be the empty tensor if no weights are used. // binary_count: bool; whether to output the number of occurrences of each value or 1.
// binary_output: Whether to output the number of occurrences of each value or 1. // output_type: dtype; dtype of the output values tensor.
// //
// Returns: // Returns:
// output_indices: Indices tensor for the resulting sparse tensor object. // output_indices: int64; indices tensor for the resulting sparse tensor object.
// output_values: Values tensor for the resulting sparse tensor object. // output_values: int64 or float32; values tensor for the resulting sparse tensor object.
// output_dense_shape: Shape tensor for the resulting sparse tensor object. // END
// }
// out_arg {
// name: "output_dense_shape"
// description: <<END
// int64; shape tensor for the resulting sparse tensor object.
// END // END
// } // }
// attr { // attr {
// name: "T" // name: "T"
// description: <<END // description: <<END
// Dtype of the input values tensor. // dtype; dtype of the input values tensor.
func RaggedCountSparseOutput(scope *Scope, splits tf.Output, values tf.Output, weights tf.Output, binary_output bool, optional ...RaggedCountSparseOutputAttr) (output_indices tf.Output, output_values tf.Output, output_dense_shape tf.Output) { // output_dense_shape
func RaggedCountSparseOutput(scope *Scope, splits tf.Output, values tf.Output, weights tf.Output, binary_count bool, output_type tf.DataType, optional ...RaggedCountSparseOutputAttr) (output_indices tf.Output, output_values tf.Output, output_dense_shape tf.Output) {
if scope.Err() != nil { if scope.Err() != nil {
return return
} }
attrs := map[string]interface{}{"binary_output": binary_output} attrs := map[string]interface{}{"binary_count": binary_count, "output_type": output_type}
for _, a := range optional { for _, a := range optional {
a(attrs) a(attrs)
} }
@ -13700,7 +13706,7 @@ type SparseCountSparseOutputAttr func(optionalAttr)
// SparseCountSparseOutputMinlength sets the optional minlength attribute to value. // SparseCountSparseOutputMinlength sets the optional minlength attribute to value.
// //
// value: Minimum value to count. Can be set to -1 for no minimum. // value: int32; minimum value to count. Can be set to -1 for no minimum.
// If not specified, defaults to -1 // If not specified, defaults to -1
// //
// REQUIRES: value >= -1 // REQUIRES: value >= -1
@ -13712,7 +13718,7 @@ func SparseCountSparseOutputMinlength(value int64) SparseCountSparseOutputAttr {
// SparseCountSparseOutputMaxlength sets the optional maxlength attribute to value. // SparseCountSparseOutputMaxlength sets the optional maxlength attribute to value.
// //
// value: Maximum value to count. Can be set to -1 for no maximum. // value: int32; maximum value to count. Can be set to -1 for no maximum.
// If not specified, defaults to -1 // If not specified, defaults to -1
// //
// REQUIRES: value >= -1 // REQUIRES: value >= -1
@ -13727,22 +13733,22 @@ func SparseCountSparseOutputMaxlength(value int64) SparseCountSparseOutputAttr {
// Counts the number of times each value occurs in the input. // Counts the number of times each value occurs in the input.
// //
// Arguments: // Arguments:
// indices: Tensor containing the indices of the sparse tensor to count. // indices: int64; Tensor containing the indices of the sparse tensor to count.
// values: Tensor containing values of the sparse tensor to count. // values: int32 or int64; Tensor containing values of the sparse tensor to count.
// dense_shape: Tensor containing the dense shape of the sparse tensor to count. // dense_shape: int64; Tensor containing the dense shape of the sparse tensor to count.
// weights: A Tensor of the same shape as indices containing per-index weight values. // weights: float32; Optional rank 1 Tensor (shape=[max_values]) with weights for each count value.
// May also be the empty tensor if no weights are used. // binary_count: bool; whether to output the number of occurrences of each value or 1.
// binary_output: Whether to output the number of occurrences of each value or 1. // output_type: dtype; dtype of the output values tensor.
// //
// Returns: // Returns:
// output_indices: Indices tensor for the resulting sparse tensor object. // output_indices: int64; indices tensor for the resulting sparse tensor object.
// output_values: Values tensor for the resulting sparse tensor object. // output_values: int64 or float32; values tensor for the resulting sparse tensor object.
// output_dense_shape: Shape tensor for the resulting sparse tensor object. // output_dense_shape: int64; shape tensor for the resulting sparse tensor object.
func SparseCountSparseOutput(scope *Scope, indices tf.Output, values tf.Output, dense_shape tf.Output, weights tf.Output, binary_output bool, optional ...SparseCountSparseOutputAttr) (output_indices tf.Output, output_values tf.Output, output_dense_shape tf.Output) { func SparseCountSparseOutput(scope *Scope, indices tf.Output, values tf.Output, dense_shape tf.Output, weights tf.Output, binary_count bool, output_type tf.DataType, optional ...SparseCountSparseOutputAttr) (output_indices tf.Output, output_values tf.Output, output_dense_shape tf.Output) {
if scope.Err() != nil { if scope.Err() != nil {
return return
} }
attrs := map[string]interface{}{"binary_output": binary_output} attrs := map[string]interface{}{"binary_count": binary_count, "output_type": output_type}
for _, a := range optional { for _, a := range optional {
a(attrs) a(attrs)
} }