Update ops-related pbtxt files.

PiperOrigin-RevId: 351639726
Change-Id: Ie49e486b9ec9a0d71291c3367a46e593e42bf56d
This commit is contained in:
A. Unique TensorFlower 2021-01-13 12:17:49 -08:00 committed by TensorFlower Gardener
parent bb61a94322
commit 0c521a8f57
2 changed files with 126 additions and 0 deletions

View File

@ -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
}

View File

@ -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 {