Home
last modified time | relevance | path

Searched refs:DevicePut (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Djax_jit.h168 xla::StatusOr<DevicePutResult> DevicePut(pybind11::handle arg,
Djax_jit.cc740 xla::StatusOr<DevicePutResult> DevicePut(pybind11::handle arg, in DevicePut() function
1010 DevicePut(arg, data_device, jax_enable_x64, pyclient)); in ConvertArgsToBuffers()
1239 DevicePut(obj, to_device.contents, jax_enable_x64, *pyclient); in BuildJaxjitSubmodule()