Searched refs:pyclient (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | jax_jit.cc | 571 bool jax_enable_x64, xla::PyClient& pyclient) { in HandleBool() argument
573 h, pyclient.pjrt_client(), to_device), in HandleBool()
578 bool jax_enable_x64, xla::PyClient& pyclient) { in HandleInt() argument
581 obj, pyclient.pjrt_client(), to_device), in HandleInt()
585 obj, pyclient.pjrt_client(), to_device), in HandleInt()
594 xla::PyClient& pyclient) { in HandleFloat() argument
597 h, pyclient.pjrt_client(), to_device), in HandleFloat()
601 h, pyclient.pjrt_client(), to_device), in HandleFloat()
610 xla::PyClient& pyclient) { in HandleComplex() argument
623 ValueOrThrow(pyclient.pjrt_client()->BufferFromHostBuffer( in HandleComplex()
[all …]
|
| D | jax_jit.h | 171 xla::PyClient& pyclient);
|