Home
last modified time | relevance | path

Searched refs:GetPjRtHostMemoryForDeviceManager (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_executable.cc104 client()->pjrt_client()->GetPjRtHostMemoryForDeviceManager(); in ExecuteInternal()
202 client()->pjrt_client()->GetPjRtHostMemoryForDeviceManager(); in ExecuteShardedOnLocalDevicesInternal()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dhost_callback.h88 client->GetPjRtHostMemoryForDeviceManager()) {} in HostCallbackContext()
H A Dpjrt_client.h780 PjRtHostMemoryForDeviceManager* GetPjRtHostMemoryForDeviceManager() const { in GetPjRtHostMemoryForDeviceManager() function