compute_aggregation_axes

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

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.