Status AllocateShapedBuffer(DataType dtype, const xla::Shape& on_device_shape,
                              xla::LocalClient* client, int device_ordinal);

  // Some Tensors can have complex on-device shapes, including tuple shapes. To
  // manage the memory for these tensors a ShapedBuffer may be required.

  // Return true if this XlaTensor contains a ShapedBuffer.
  bool has_shaped_buffer() const { return shaped_buffer_.has_value(); }

// checking where a certain attribute gets lost. Note that we use a semicolon
// instead of comma as the separator to allow strings that contain commas (which
// frequently appear in MLIR). The strings can contain any characters (including
// spaces) except semicolons.
//
// Example: Setting the environment variables
// `MLIR_BRIDGE_LOG_PASS_FILTER="LegalizeTF;Canonicalizer"` and
// `MLIR_BRIDGE_LOG_STRING_FILTER="my_string"` will dump IR only for invocations