Searched refs:HostCommmandHandler (Results 1 – 1 of 1) sorted by relevance
69 using HostCommmandHandler = xla::TpuExecutable::HostCommandHandler; typedef in tensorflow::__anon5955583b0111::HostTransferManager73 xla::StatusOr<HostCommmandHandler> Initialize(81 xla::StatusOr<HostTransferManager::HostCommmandHandler>85 return HostCommmandHandler([](uint32, int64) { in Initialize()433 TF_ASSIGN_OR_RETURN(HostTransferManager::HostCommmandHandler handler, in TPUExecute()