Searched refs:host_memory_for_device_manager_ (Results 1 – 3 of 3) sorted by relevance
94 host_memory_for_device_manager_(host_memory_for_device_manager), in HostCallbackContext()98 CHECK(host_memory_for_device_manager_); in HostCallbackContext()115 PjRtHostMemoryForDeviceManager* host_memory_for_device_manager_ = nullptr; variable
33 TF_CHECK_OK(host_memory_for_device_manager_->ToHostLayout( in OnSend()95 auto statusor_linearized = host_memory_for_device_manager_->ToDeviceLayout( in Receive()
431 : host_memory_for_device_manager_( in PjRtClient()781 return host_memory_for_device_manager_.get(); in GetPjRtHostMemoryForDeviceManager()786 host_memory_for_device_manager_;