Home
last modified time | relevance | path

Searched defs:num_devices_per_task (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dcollective.h73 std::unordered_map<string, int32> num_devices_per_task; member
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util_test.cc259 int num_devices_per_task) { in MakeDeviceSet()