mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-16 07:35:16 -05:00
1.3 KiB
1.3 KiB
module concrete.fhe.extensions.univariate
Declaration of univariate function.
function univariate
univariate(
function: Callable[[Any], Any],
outputs: Optional[BaseDataType, Type[ScalarAnnotation]] = None
) → Callable[[Union[Tracer, Any]], Union[Tracer, Any]]
Wrap a univariate function so that it is traced into a single generic node.
Args: function (Callable[[Any], Any]): univariate function to wrap
outputs (Optional[Union[BaseDataType, Type[ScalarAnnotation]]], default = None): data type of the result, unused during compilation, required for direct definition
Returns: CallableUnion[Tracer, Any, Union[Tracer, Any]]: another univariate function that can be called with a Tracer as well