blob: 08414b3e6818e1d08b938966e4fb947b7901d26e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
op {
visibility: HIDDEN
graph_op_name: "ReduceDataset"
in_arg {
name: "input_dataset"
description: <<END
A variant tensor representing the input dataset.
END
}
in_arg {
name: "initial_state"
description: <<END
A nested structure of tensors, representing the initial state of the
transformation.
END
}
attr {
name: "f"
description: <<END
A function that maps `(old_state, input_element)` to `new_state`. It must take
two arguments and return a nested structures of tensors. The structure of
`new_state` must match the structure of `initial_state`.
END
}
summary: "Reduces the input dataset to a singleton using a reduce function."
}
|