Home
last modified time | relevance | path

Searched refs:kNumDeviceToDeviceStreams (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dlocal_device_state.cc58 device_to_device_streams_.reserve(kNumDeviceToDeviceStreams); in LocalDeviceState()
59 for (int i = 0; i < kNumDeviceToDeviceStreams; ++i) { in LocalDeviceState()
Dlocal_device_state.h183 static constexpr int kNumDeviceToDeviceStreams = 4; variable
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc236 static constexpr int kNumDeviceToDeviceStreams = 4; in XlaDevice() local
237 device_to_device_streams_.resize(kNumDeviceToDeviceStreams); in XlaDevice()