Searched refs:WrapWithClient (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.cc | 43 devices.push_back(WrapWithClient(shared_from_this(), device)); in Devices()
52 devices.push_back(WrapWithClient(shared_from_this(), device)); in LocalDevices()
62 buffers.push_back(WrapWithClient(shared_from_this(), buffer)); in LiveBuffers()
81 result[r][p] = WrapWithClient(shared_from_this(), device); in GetDefaultDeviceAssignment()
97 result.push_back(WrapWithClient(shared_from_this(), device)); in GetDefaultDeviceAssignment1D()
|
| D | py_client.h | 80 ClientAndPtr<T> WrapWithClient(std::shared_ptr<PyClient> client, T* contents) { in WrapWithClient() function
|
| D | outfeed_receiver_py.cc | 109 callback_python_(WrapWithClient<PjRtDevice>(*it, device), consumer_id, in Callback()
|
| D | py_executable.cc | 65 devices.push_back(WrapWithClient(client_, device)); in AddressableDevices()
|
| D | py_buffer.cc | 68 return WrapWithClient(client_, buffer_->device()); in device()
|