Skip to main content

Module extension

Module extension 

Source
Expand description

A minimal extension-owned operation for the external scalar.

The operation is an ExtensionOp family whose payload, numerical body, and output construction all live in this crate. The runtime reaches it through a registered ExtensionModule and prepared execution, and the input travels as a runtime Tensor that carries a caller-owned payload.

Structs§

Df64Einsum
A matrix contraction in the external scalar, written in ordinary einsum notation.
Df64EinsumJvp
The forward tangent of a two-input contraction.
Df64EinsumVjp
The adjoint of a two-input contraction.
Df64Expand
Broadcast a scalar external value to a declared shape.
Df64FromF64
Widen a preset f64 tensor into the externally defined scalar.
Df64Qr
Reduced QR factorization of a real square or tall matrix.
Df64QrJvp
Forward-mode tangent of the reduced QR factorization.
Df64QrVjp
Reverse-mode adjoint of the reduced QR factorization.
Df64ToF64
Narrow the externally defined scalar into a preset f64 tensor.
Df64Total
Family identifier of the extension-owned scalar broadcast. Total sum of an externally defined scalar tensor.

Constants§

DF64_OPS_FAMILY
Family identifier of the contribution’s externally defined operations.
DF64_SCALAR
The externally defined element type of the contribution’s scalar.
DF64_SCALAR_IDENTITY
Canonical identity of the externally defined Df64 scalar.

Functions§

apply_total
Run Df64Total on an eagerly held external tensor.
module
The module a downstream application installs to enable Df64Total.
module_for_engine
The module a downstream application installs when it composes more than one CPU backend in one runtime.