aihwkit.simulator.tiles.functions module

Autograd functions for aihwkit.

class aihwkit.simulator.tiles.functions.AnalogFunction(*args, **kwargs)[source]

Bases: Function

Function for analog functions.

static backward(ctx, grad_output)[source]

Execute the backward pass in the analog tile.

Parameters:

  • ctx (Any)

  • grad_output (Tensor)

Return type:

Tuple[Tensor | None, Tensor | None, Tensor | None, Tensor | None, Tensor | None]

static forward(ctx, analog_ctx, analog_tile, input_, shared_weights=None, is_test=False)[source]

Execute the forward pass in the analog tile. Note: Indexed versions can used when analog_ctx.use_indexed is set to True.

Parameters:

  • ctx (Any)

  • analog_ctx (AnalogContext)

  • analog_tile (Any)

  • input_ (Tensor)

  • shared_weights (Tensor | None)

  • is_test (bool)

Return type:

Tensor