Searched refs:host_to_device (Results 1 – 3 of 3) sorted by relevance
36 // Metadata about each host_to_device transfer37 repeated HostTransferMetadata host_to_device = 2; field
140 se::Stream* host_to_device = nullptr; member
250 group->host_to_device = new se::Stream(executor); in GetOrCreate()251 group->host_to_device->Init(); in GetOrCreate()253 << "] = " << group->host_to_device; in GetOrCreate()391 i, streams_.back()->compute, streams_.back()->host_to_device, in Init()