Use xla::ifrt::AttributeMap for cost analysis results
This is for consistency with other attribute maps in IFRT. Also, using AttributeMap makes it possible to provide stable serialization, e.g., for IFRT Proxy.
JAX Python binding still uses xla::PjRtValueType since pybind/nanobind doesn't recognize xla::ifrt::AttributeMap::Value as Python native types.
Use
xla::ifrt::AttributeMap
for cost analysis resultsThis is for consistency with other attribute maps in IFRT. Also, using
AttributeMap
makes it possible to provide stable serialization, e.g., for IFRT Proxy.JAX Python binding still uses
xla::PjRtValueType
since pybind/nanobind doesn't recognizexla::ifrt::AttributeMap::Value
as Python native types.