Update ops-related pbtxt files.

PiperOrigin-RevId: 298496614
Change-Id: I9829b9c660eaa7902046bb2686dd99e09de44671
This commit is contained in:
A. Unique TensorFlower 2020-03-02 18:18:35 -08:00 committed by TensorFlower Gardener
parent 72e7964b6a
commit 9cef7b78b0
2 changed files with 246 additions and 0 deletions

View File

@ -0,0 +1,123 @@
op {
name: "RaggedCross"
input_arg {
name: "ragged_values"
type_list_attr: "ragged_values_types"
}
input_arg {
name: "ragged_row_splits"
type_list_attr: "ragged_splits_types"
}
input_arg {
name: "sparse_indices"
type: DT_INT64
number_attr: "Nsparse"
}
input_arg {
name: "sparse_values"
type_list_attr: "sparse_values_types"
}
input_arg {
name: "sparse_shape"
type: DT_INT64
number_attr: "Nsparse"
}
input_arg {
name: "dense_inputs"
type_list_attr: "dense_types"
}
output_arg {
name: "output_values"
type_attr: "out_values_type"
}
output_arg {
name: "output_row_splits"
type_attr: "out_row_splits_type"
}
attr {
name: "Nsparse"
type: "int"
has_minimum: true
}
attr {
name: "input_order"
type: "string"
}
attr {
name: "hashed_output"
type: "bool"
}
attr {
name: "num_buckets"
type: "int"
has_minimum: true
}
attr {
name: "hash_key"
type: "int"
}
attr {
name: "ragged_values_types"
type: "list(type)"
has_minimum: true
allowed_values {
list {
type: DT_INT64
type: DT_STRING
}
}
}
attr {
name: "ragged_splits_types"
type: "list(type)"
has_minimum: true
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "sparse_values_types"
type: "list(type)"
has_minimum: true
allowed_values {
list {
type: DT_INT64
type: DT_STRING
}
}
}
attr {
name: "dense_types"
type: "list(type)"
has_minimum: true
allowed_values {
list {
type: DT_INT64
type: DT_STRING
}
}
}
attr {
name: "out_values_type"
type: "type"
allowed_values {
list {
type: DT_INT64
type: DT_STRING
}
}
}
attr {
name: "out_row_splits_type"
type: "type"
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}

View File

@ -32675,6 +32675,129 @@ op {
}
}
}
op {
name: "RaggedCross"
input_arg {
name: "ragged_values"
type_list_attr: "ragged_values_types"
}
input_arg {
name: "ragged_row_splits"
type_list_attr: "ragged_splits_types"
}
input_arg {
name: "sparse_indices"
type: DT_INT64
number_attr: "Nsparse"
}
input_arg {
name: "sparse_values"
type_list_attr: "sparse_values_types"
}
input_arg {
name: "sparse_shape"
type: DT_INT64
number_attr: "Nsparse"
}
input_arg {
name: "dense_inputs"
type_list_attr: "dense_types"
}
output_arg {
name: "output_values"
type_attr: "out_values_type"
}
output_arg {
name: "output_row_splits"
type_attr: "out_row_splits_type"
}
attr {
name: "Nsparse"
type: "int"
has_minimum: true
}
attr {
name: "input_order"
type: "string"
}
attr {
name: "hashed_output"
type: "bool"
}
attr {
name: "num_buckets"
type: "int"
has_minimum: true
}
attr {
name: "hash_key"
type: "int"
}
attr {
name: "ragged_values_types"
type: "list(type)"
has_minimum: true
allowed_values {
list {
type: DT_INT64
type: DT_STRING
}
}
}
attr {
name: "ragged_splits_types"
type: "list(type)"
has_minimum: true
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "sparse_values_types"
type: "list(type)"
has_minimum: true
allowed_values {
list {
type: DT_INT64
type: DT_STRING
}
}
}
attr {
name: "dense_types"
type: "list(type)"
has_minimum: true
allowed_values {
list {
type: DT_INT64
type: DT_STRING
}
}
}
attr {
name: "out_values_type"
type: "type"
allowed_values {
list {
type: DT_INT64
type: DT_STRING
}
}
}
attr {
name: "out_row_splits_type"
type: "type"
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "RaggedGather"
input_arg {