Searched defs:pyclient (Results 1 – 1 of 1) sorted by relevance
571 bool jax_enable_x64, xla::PyClient& pyclient) { in HandleBool()578 bool jax_enable_x64, xla::PyClient& pyclient) { in HandleInt()594 xla::PyClient& pyclient) { in HandleFloat()610 xla::PyClient& pyclient) { in HandleComplex()643 xla::PyClient& pyclient) { in HandleDeviceArray()665 xla::PyClient& pyclient) { in HandleBufferFromPyval()675 bool jax_enable_x64, xla::PyClient& pyclient) { in HandleNpBool()688 bool jax_enable_x64, xla::PyClient& pyclient) { in HandleUint64()713 xla::PyClient& pyclient) { in HandleNdarray()743 xla::PyClient& pyclient) { in DevicePut()[all …]