experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Document the iota_dimension attribute for kIota

Browse Source 
PiperOrigin-RevId: 223520378
This commit is contained in:
David Majnemer 2018-11-30 08:37:01 -08:00 committed by TensorFlower Gardener
parent d9833ee204
commit f5291176d4
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/compiler/xla/g3doc/operation_semantics.md
View File

@ -1466,10 +1466,11 @@ Builds a constant literal on device rather than a potentially large host
transfer. Creates a rank 1 array of values starting at zero and incrementing by
one.
Arguments | Type | Semantics
--------- | --------------- | ------------------------------------
`type` | `PrimitiveType` | type U
`size` | `int64` | The number of elements in the array.
Arguments | Type | Semantics
---------------- | --------------- | ------------------------------------
`type` | `PrimitiveType` | type U
`size` | `int64` | The number of elements in the array.
`iota_dimension` | `int64` | The dimension to increment along.
## Map