yobx.tensorflow.ops.xla_call_module_helper#

Helper utilities for the xla_call_module StableHLO→ONNX converter.

This module contains low-level extraction utilities (bracket/brace balancing, function-body extraction, dtype/reduce-op maps) that have no dependency on TensorFlow, JAX, or any ONNX graph-builder state.

Functions whose names contain parse (e.g. _parse_tensor_type(), _parse_dense_value(), _parse_body()) live in xla_call_module_parsing instead.

This Page