code: def _cpp_device_put(value, device):
      aval = api_util.shaped_abstractify(value)
      return pxla.batched_device_put(
          aval,
          jax.sharding.SingleDeviceSharding(device),
          [value],
          [device],
      )
  
imports: import inspect
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import dtypes
  from jax import numpy
  from jax._src import api_util
  from jax._src import config
  from jax._src import core
  from jax._src import lib
  from jax._src import test_util
  from jax._src.interpreters import pxla
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: _cpp_device_put(value, device) and returns: pxla