Searched refs:device_to_device_stream (Results 1 – 4 of 4) sorted by relevance
35 gtl::InlinedVector<se::Stream*, 4> device_to_device_stream) in GPUDeviceContext() argument40 device_to_device_stream_(device_to_device_stream) {} in GPUDeviceContext()47 se::Stream* device_to_device_stream(int index) const { in device_to_device_stream() function
78 se::Stream* device_to_device_stream(int index) const { in device_to_device_stream() function
264 return device_to_device_stream(stream); in GetDeviceToDeviceStream()
203 ->device_to_device_stream(dev_to_dev_stream_index); in DeviceToDeviceCopy()