0

我正在比较 XLAxla::Map和元素操作,我想弄清楚何时使用它们。

API

快速浏览一下,似乎所有对标量的计算也可以定义为任意形状,因为所有相关的操作都有逐元素的变体(EqSinAdd)。我不清楚Map除了替代 API 还提供什么。或者只是为了让我们可以使用预先构建的标量计算,在这种情况下它与 有何不同Call

表现

是否xla::Map具有与元素操作相同的性能特征?也就是说,在伪代码中,是

Map(x, Abs)
Map(x, compose(Neg, Abs))
Map({x, y}, Eq)

优化至少大致与

Abs(x)
Neg(Abs(x))
Eq(x, y)
4

0 回答 0