blob: afa14e8e8a8b2039afd288aa238f5dbd650e6b0a [file] [log] [blame]
op {
name: "RaggedGather"
input_arg {
name: "params_nested_splits"
type: DT_INT64
number_attr: "PARAMS_RAGGED_RANK"
}
input_arg {
name: "params_dense_values"
type_attr: "Tvalues"
}
input_arg {
name: "indices"
type_attr: "Tindices"
}
output_arg {
name: "output_nested_splits"
type: DT_INT64
number_attr: "OUTPUT_RAGGED_RANK"
}
output_arg {
name: "output_dense_values"
type_attr: "Tvalues"
}
attr {
name: "Tvalues"
type: "type"
}
attr {
name: "Tindices"
type: "type"
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "PARAMS_RAGGED_RANK"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "OUTPUT_RAGGED_RANK"
type: "int"
has_minimum: true
}
}
op {
name: "RaggedGather"
input_arg {
name: "params_nested_splits"
type_attr: "Tsplits"
number_attr: "PARAMS_RAGGED_RANK"
}
input_arg {
name: "params_dense_values"
type_attr: "Tvalues"
}
input_arg {
name: "indices"
type_attr: "Tindices"
}
output_arg {
name: "output_nested_splits"
type_attr: "Tsplits"
number_attr: "OUTPUT_RAGGED_RANK"
}
output_arg {
name: "output_dense_values"
type_attr: "Tvalues"
}
attr {
name: "Tvalues"
type: "type"
}
attr {
name: "Tindices"
type: "type"
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "Tsplits"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "PARAMS_RAGGED_RANK"
type: "int"
has_minimum: true
minimum: 1
}
attr {
name: "OUTPUT_RAGGED_RANK"
type: "int"
has_minimum: true
}
}