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.cc54 device_to_device_streams_.reserve(kNumDeviceToDeviceStreams); in LocalDeviceState()
55 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.cc224 static constexpr int kNumDeviceToDeviceStreams = 4; in XlaDevice() local
225 device_to_device_streams_.resize(kNumDeviceToDeviceStreams); in XlaDevice()