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
47 device_to_device_stream_(device_to_device_stream), in GPUDeviceContext()60 return device_to_device_stream_[index % device_to_device_stream_.size()]; in device_to_device_stream()99 gtl::InlinedVector<se::Stream*, 4> device_to_device_stream_; variable