op { | |
name: "ShardDataset" | |
input_arg { | |
name: "input_dataset" | |
type: DT_VARIANT | |
} | |
input_arg { | |
name: "num_shards" | |
type: DT_INT64 | |
} | |
input_arg { | |
name: "index" | |
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 | |
} | |
} | |
op { | |
name: "ShardDataset" | |
input_arg { | |
name: "input_dataset" | |
type: DT_VARIANT | |
} | |
input_arg { | |
name: "num_shards" | |
type: DT_INT64 | |
} | |
input_arg { | |
name: "index" | |
type: DT_INT64 | |
} | |
output_arg { | |
name: "handle" | |
type: DT_VARIANT | |
} | |
attr { | |
name: "require_non_empty" | |
type: "bool" | |
default_value { | |
b: false | |
} | |
} | |
attr { | |
name: "output_types" | |
type: "list(type)" | |
has_minimum: true | |
minimum: 1 | |
} | |
attr { | |
name: "output_shapes" | |
type: "list(shape)" | |
has_minimum: true | |
minimum: 1 | |
} | |
} |