// `logical_core` is the index of the TPU core in [0, num_cores_per_replica).
// `device_ordinal` is the index of the TPU core relative to its host.
LogicalResult GetDeviceOrdinal(const std::optional<DictionaryAttr>& devices,
                               const unsigned replica_id,
                               const uint64_t logical_core,

xla::ExecuteOptions GetPjRtExecuteOptions(
    const DeviceType& device_type,
    absl::flat_hash_set<int> non_donatable_input_indices);

// Returns the device ordinal from the parsed name of the device.
int GetDeviceOrdinal(const DeviceBase* device);

// Returns the device type from the OpKernelContext.
DeviceType GetDeviceType(OpKernelContext* ctx);

// Runs `executable` and populates the outputs in `ctx`. `inputs` and

  return GetAbsDeviceStr(*this, getDeviceOrdinal());
}

std::optional<std::string>
EnqueueTPUEmbeddingBatchOp::GetResourceInstanceStr() {
  return GetAbsDeviceStr(*this, getDeviceOrdinal());
}

std::optional<std::string>
EnqueueTPUEmbeddingIntegerBatchOp::GetResourceInstanceStr() {
  return GetAbsDeviceStr(*this, getDeviceOrdinal());
}

std::optional<std::string>