mirror of
https://github.com/zebrajr/tensorflow.git
synced 2025-12-08 07:38:39 +01:00
Update ops-related pbtxt files.
PiperOrigin-RevId: 158413455
This commit is contained in:
parent
1e90b78e93
commit
50b2f951cb
|
|
@ -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 {
|
||||
|
|
|
|||
|
|
@ -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 {
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user