Primitive Keys

autoform.extend.format_p = format

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.concat_p = concat

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.match_p = match

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.lm_call_p = lm_call

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.lm_schema_call_p = lm_schema_call

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.checkpoint_p = checkpoint

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.stop_gradient_p = stop_gradient

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.switch_p = switch

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.while_loop_p = while_loop

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.gather_p = gather

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.depends_p = depends

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.batch_call_p = batch_call

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.pushforward_call_p = pushforward_call

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")

autoform.extend.pullback_call_p = pullback_call

Primitive operation key used by interpreter rule registries.

A primitive has no behavior by itself. Runtime, abstract, batching, and AD behavior are attached by registering rules keyed by the Prim instance.

Parameters:

name – The name of the primitive.

Example

>>> import autoform.extend as afe
>>> add = afe.Prim("add")