blob: 6b54fa1307000e6bbbe3508d52fc91cdbb6c09b6 (
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
27
28
29
30
|
op {
graph_op_name: "MultiDeviceIteratorInit"
in_arg {
name: "dataset"
description: <<END
Dataset to be iterated upon.
END
}
in_arg {
name: "multi_device_iterator"
description: <<END
A MultiDeviceIteratorResource.
END
}
in_arg {
name: "max_buffer_size"
description: <<END
The maximum size of the host side per device buffer to keep.
END
}
out_arg {
name: "incarnation_id"
description: <<END
An int64 indicating which incarnation of the MultiDeviceIterator
is running.
END
}
summary: "Initializes the multi device iterator with the given dataset."
visibility: HIDDEN
}
|
