blob: d158f4b50289e56b7b4cbd9192a6d7dba260ecf6 (
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
op {
graph_op_name: "MapDefun"
visibility: HIDDEN
in_arg {
name: "arguments"
description: <<END
A list of tensors whose types are `Targuments`, corresponding to the inputs
the function should be mapped over.
END
}
in_arg {
name: "captured_inputs"
description: <<END
A list of tensors whose types are `Tcaptured`, corresponding to the captured
inputs of the defun.
END
}
out_arg {
name: "output"
description: <<END
A list of output tensors whose types are `output_types` and whose dimensions
0 are the same as the dimensions 0 of the tensors in `arguments`, and whose
remaining dimensions correspond to those in `output_shapes`.
END
}
attr {
name: "Targuments"
description: "A list of types."
}
attr {
name: "Tcaptured"
description: "A list of types."
}
attr {
name: "output_types"
description: "A list of types."
}
attr {
name: "output_shapes"
description: "A list of shapes."
}
summary: <<END
Maps a function on the list of tensors unpacked from arguments on dimension 0.
END
}
|