Efficient storage of broadcasted tensor

9 views
Skip to first unread message

Joel Berkeley

unread,
8:37 PM (1 hour ago) 8:37 PM
to OpenXLA Discuss
Hi,

If I broadcast a tensor to match shapes, will XLA typically notice that and not duplicate my data, to save space (and possibly time)? For example, in
```
x = [1, 2, 3]
y = rand([100, 3])
```
I can do `broadcast x + y`, but I don't want to duplicate the data in x.
Reply all
Reply to author
Forward
0 new messages