Searched defs:host_memory_for_device_manager (Results 1 – 4 of 4) sorted by relevance
103 auto* host_memory_for_device_manager = in ExecuteInternal() local201 auto* host_memory_for_device_manager = in ExecuteShardedOnLocalDevicesInternal() local
92 PjRtHostMemoryForDeviceManager* host_memory_for_device_manager) in HostCallbackContext()
103 PjRtHostMemoryForDeviceManager* host_memory_for_device_manager, in CreateHostCallbackStateAndAppendSendRecvCallbacks()
430 host_memory_for_device_manager) in PjRtClient()