Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1954 const int num_addressable_devices = addressable_devices_.size(); in Execute() local
1956 if (argument_handles.size() != num_addressable_devices) { in Execute()
1960 argument_handles.size(), num_addressable_devices, num_replicas(), in Execute()
1967 << " num_addressable_devices=" << num_addressable_devices; in Execute()
1969 num_addressable_devices); in Execute()
1970 if (num_addressable_devices == 1) { in Execute()
1979 int running = num_addressable_devices; in Execute()
1983 for (int i = 0; i < num_addressable_devices; ++i) { in Execute()
2034 num_addressable_devices); in Execute()
2035 for (int i = 0; i < num_addressable_devices; ++i) { in Execute()
[all …]