experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update ops-related pbtxt files.

Browse Source 
PiperOrigin-RevId: 309145167
Change-Id: I893d63daef7e3668e7a978bf6a4c3a70e6a42f12
This commit is contained in:
A. Unique TensorFlower 2020-04-29 19:19:34 -07:00 committed by TensorFlower Gardener
parent 35a7fe77a2
commit b5f954b27d
3 changed files with 155 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
tensorflow/core/ops/compat/ops_history_v2/ShuffleAndRepeatDataset.pbtxt
View File

@ -37,3 +37,49 @@ op {
minimum: 1 minimum: 1
} }
} }
op {
name: "ShuffleAndRepeatDataset"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "buffer_size"
type: DT_INT64
}
input_arg {
name: "seed"
type: DT_INT64
}
input_arg {
name: "seed2"
type: DT_INT64
}
input_arg {
name: "count"
type: DT_INT64
}
output_arg {
name: "handle"
type: DT_VARIANT
}
attr {
name: "output_types"
type: "list(type)"
has_minimum: true
minimum: 1
}
attr {
name: "output_shapes"
type: "list(shape)"
has_minimum: true
minimum: 1
}
attr {
name: "reshuffle_each_iteration"
type: "bool"
default_value {
b: true
}
}
}

51
tensorflow/core/ops/compat/ops_history_v2/ShuffleAndRepeatDatasetV2.pbtxt Normal file
View File

@ -0,0 +1,51 @@
op {
name: "ShuffleAndRepeatDatasetV2"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "buffer_size"
type: DT_INT64
}
input_arg {
name: "seed"
type: DT_INT64
}
input_arg {
name: "seed2"
type: DT_INT64
}
input_arg {
name: "count"
type: DT_INT64
}
input_arg {
name: "seed_generator"
type: DT_RESOURCE
}
output_arg {
name: "handle"
type: DT_VARIANT
}
attr {
name: "reshuffle_each_iteration"
type: "bool"
default_value {
b: true
}
}
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
}

58
tensorflow/core/ops/ops.pbtxt
View File

@ -42275,6 +42275,64 @@ op {
has_minimum: true has_minimum: true
minimum: 1 minimum: 1
} }
attr {
name: "reshuffle_each_iteration"
type: "bool"
default_value {
b: true
}
}
}
op {
name: "ShuffleAndRepeatDatasetV2"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "buffer_size"
type: DT_INT64
}
input_arg {
name: "seed"
type: DT_INT64
}
input_arg {
name: "seed2"
type: DT_INT64
}
input_arg {
name: "count"
type: DT_INT64
}
input_arg {
name: "seed_generator"
type: DT_RESOURCE
}
output_arg {
name: "handle"
type: DT_VARIANT
}
attr {
name: "reshuffle_each_iteration"
type: "bool"
default_value {
b: true
}
}
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 { op {
name: "ShuffleDataset" name: "ShuffleDataset"