ExecutionObserver

Observer that can tap into execution events produced by an ExecutionContext.

Implementations can be used to collect metrics, trace operations or build higher-level benchmark tooling.

Functions

open fun onOpEnd(context: ExecutionContext, opName: String, result: Any?)

Called after a successful operation. result can be a single tensor, a collection of tensors, or any backend specific payload.

Called if the backend throws while evaluating an operation.

open fun onOpStart(context: ExecutionContext, opName: String, inputs: List<Tensor<*, *>>)

Called before the backend executes an operation.

Called whenever a tensor is materialised inside the context (e.g. as the result of an operation or via factory helpers).