Skip to contents

Create a custom call operation that invokes an external function via the FFI (Foreign Function Interface) API.

Note that the attributes called_computations and output_operand_aliases are not implemented yet.

Usage

hlo_custom_call(
  ...,
  call_target_name,
  api_version = 4L,
  has_side_effect,
  backend_config = NULL,
  output_types = NULL
)

Arguments

...

(FuncValue)
Input values to pass to the custom call.

call_target_name

(character(1))
The name of the registered custom function to call.

api_version

(integer(1))
The API version. Default is 4.

has_side_effect

(logical(1))
Whether the custom call has side effects.

backend_config

(CustomOpBackendConfig | NULL)
Optional backend configuration.

output_types

(list of ValueType | NULL)
The output types of the custom call. Default is NULL (no outputs).

Value

(FuncValue | list() | NULL)
The output value(s), or NULL for side-effect only calls.