Update ops-related pbtxt files.

PiperOrigin-RevId: 306952881
Change-Id: Id7a63a053054e7532eba1c52ea0863b6a5d0020b
This commit is contained in:
A. Unique TensorFlower 2020-04-16 17:19:20 -07:00 committed by TensorFlower Gardener
parent 6b0ab72dfc
commit 5d192714dd
2 changed files with 39 additions and 0 deletions

View File

@ -0,0 +1,23 @@
op {
name: "CreateJob"
input_arg {
name: "dataset_id"
type: DT_INT64
}
input_arg {
name: "address"
type: DT_STRING
}
input_arg {
name: "protocol"
type: DT_STRING
}
input_arg {
name: "processing_mode"
type: DT_STRING
}
output_arg {
name: "job_token"
type: DT_VARIANT
}
}

View File

@ -0,0 +1,16 @@
op {
name: "MakeDataServiceIterator"
input_arg {
name: "dataset"
type: DT_VARIANT
}
input_arg {
name: "job_token"
type: DT_VARIANT
}
input_arg {
name: "iterator"
type: DT_RESOURCE
}
is_stateful: true
}