yobx.tensorflow.ops.xla_call_module

yobx.tensorflow.ops.xla_call_module.convert_exp(g: GraphBuilderExtendedProtocol, sts: Dict[str, Any], outputs: List[str], op: Operation) → str

Convert a StableHLO XlaCallModule operation to ONNX.

Supports the following StableHLO ops: unary/binary elementwise, dot_general (matrix multiply), broadcast_in_dim, dynamic_broadcast_in_dim, constant, reduce (max/sum), call (to private functions), and comparison ops.