aeppl.abstract.MeasurableElemwise.prepare_node#

MeasurableElemwise.prepare_node(node, storage_map, compute_map, impl)#

Make any special modifications that the Op needs before doing Op.make_thunk.

This can modify the node inplace and should return nothing.

It can be called multiple time with different impl values.

Warning

It is the Op’s responsibility to not re-prepare the node when it isn’t good to do so.