blob: e2dd11d62578b43a3d424506bf7f9e6d6918e8b5 [file] [log] [blame]
op {
name: "ShuffleDatasetV2"
input_arg {
name: "input_dataset"
type: DT_VARIANT
}
input_arg {
name: "buffer_size"
type: DT_INT64
}
input_arg {
name: "seed_generator"
type: DT_RESOURCE
}
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
}
is_stateful: true
}