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::AttributeMapfor cost analysis resultsThis is for consistency with other attribute maps in IFRT. Also, using
AttributeMapmakes it possible to provide stable serialization, e.g., for IFRT Proxy.JAX Python binding still uses
xla::PjRtValueTypesince pybind/nanobind doesn't recognizexla::ifrt::AttributeMap::Valueas Python native types.