Update ops-related pbtxt files.

PiperOrigin-RevId: 158413455
This commit is contained in:
A. Unique TensorFlower 2017-06-08 10:48:09 -07:00 committed by TensorFlower Gardener
parent 1e90b78e93
commit 50b2f951cb
2 changed files with 59 additions and 0 deletions

View File

@ -3931,6 +3931,34 @@ op {
}
}
}
op {
name: "CacheDataset"
input_arg {
name: "input_dataset"
type: DT_RESOURCE
}
input_arg {
name: "filename"
type: DT_STRING
}
output_arg {
name: "handle"
type: DT_RESOURCE
}
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: "Cast"
input_arg {

View File

@ -3894,6 +3894,37 @@ op {
summary: "Calculates the CTC Loss (log probability) for each batch entry. Also calculates"
description: "the gradient. This class performs the softmax operation for you, so inputs\nshould be e.g. linear projections of outputs by an LSTM."
}
op {
name: "CacheDataset"
input_arg {
name: "input_dataset"
type: DT_RESOURCE
}
input_arg {
name: "filename"
description: "A path on the filesystem where we should cache the dataset. Note: this\nwill be a directory."
type: DT_STRING
}
output_arg {
name: "handle"
type: DT_RESOURCE
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
summary: "Creates a dataset that caches elements from `input_dataset`."
description: "A CacheDataset will iterate over the input_dataset, and store tensors. If the\ncache already exists, the cache will be used. If the cache is inappropriate\n(e.g. cannot be opened, contains tensors of the wrong shape / size), an error\nwill the returned when used."
is_stateful: true
}
op {
name: "Cast"
input_arg {