Home
last modified time | relevance | path

Searched defs:logical_device_id (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc302 const int logical_device_id = sharding.tile_assignment_devices(0); in ExtractInputsForLogicalDevices() local
399 for (const auto logical_device_id : sharding.tile_assignment_devices()) { in GetTileShardedOutputsToMerge() local
598 int logical_device_id = 0; in RemapOutputsFromLogicalDevices() local
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy.py405 def experimental_assign_to_logical_device(self, tensor, logical_device_id): argument
1094 def experimental_logical_device(self, logical_device_id): argument
1561 def experimental_logical_device(self, logical_device_id): argument
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc316 for (const auto& logical_device_id : span) { in PYBIND11_MODULE() local