Searched refs:device_to_device_stream_ (Results 1 – 2 of 2) sorted by relevance
40 device_to_device_stream_(device_to_device_stream) {} in PluggableDeviceContext()48 return device_to_device_stream_[index % device_to_device_stream_.size()]; in device_to_device_stream()82 gtl::InlinedVector<se::Stream*, 4> device_to_device_stream_; variable
46 device_to_device_stream_(device_to_device_stream) { in GPUDeviceContext()58 return device_to_device_stream_[index % device_to_device_stream_.size()]; in device_to_device_stream()94 gtl::InlinedVector<se::Stream*, 4> device_to_device_stream_; variable