Searched refs:device_to_host_streams_ (Results 1 – 2 of 2) sorted by relevance
48 device_to_host_streams_.reserve(kNumDeviceToHostStreams); in LocalDeviceState()52 device_to_host_streams_.push_back(std::move(stream)); in LocalDeviceState()110 (next_device_to_host_stream_ + 1) % device_to_host_streams_.size(); in GetDeviceToHostStream()111 return device_to_host_streams_.at(i).get(); in GetDeviceToHostStream()
178 std::vector<std::unique_ptr<se::Stream>> device_to_host_streams_; variable