Update ops-related pbtxt files.
PiperOrigin-RevId: 351639726 Change-Id: Ie49e486b9ec9a0d71291c3367a46e593e42bf56d
This commit is contained in:
parent
bb61a94322
commit
0c521a8f57
@ -0,0 +1,63 @@
|
||||
op {
|
||||
name: "DataServiceDatasetV2"
|
||||
input_arg {
|
||||
name: "dataset_id"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "processing_mode"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "address"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "protocol"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "job_name"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "consumer_index"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "num_consumers"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "max_outstanding_requests"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "iteration_counter"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
output_arg {
|
||||
name: "handle"
|
||||
type: DT_VARIANT
|
||||
}
|
||||
attr {
|
||||
name: "task_refresh_interval_hint_ms"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: -1
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "output_types"
|
||||
type: "list(type)"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "output_shapes"
|
||||
type: "list(shape)"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
@ -11218,6 +11218,69 @@ op {
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "DataServiceDatasetV2"
|
||||
input_arg {
|
||||
name: "dataset_id"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "processing_mode"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "address"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "protocol"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "job_name"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "consumer_index"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "num_consumers"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "max_outstanding_requests"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "iteration_counter"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
output_arg {
|
||||
name: "handle"
|
||||
type: DT_VARIANT
|
||||
}
|
||||
attr {
|
||||
name: "task_refresh_interval_hint_ms"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: -1
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "output_types"
|
||||
type: "list(type)"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "output_shapes"
|
||||
type: "list(shape)"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "DatasetCardinality"
|
||||
input_arg {
|
||||
|
Loading…
Reference in New Issue
Block a user