compute_aggregation_axes

plaidml.tile.compute_aggregation_axes(dims, axes=None, keepdims=False)[source]

Computes parameters for an aggregation-over-axes operation.

Parameters:
  • dims ([int or Value]) – The dimensions of the value being aggregated.
  • axes ([int], optional) – Defaults to None. The indices of the axes to aggregate over.
  • keepdims (bool, optional) – Defaults to False. Iff true, keep the aggregated axes in the output.
Returns:

(dims, axes, dict(string->string))

The resulting dimensions and axes, and a dictionary of

formatting replacements to use when building the TILE operation.