Searched refs:sharding_attr_and_index (Results 1 – 1 of 1) sorted by relevance
235 for (const auto& sharding_attr_and_index : llvm::enumerate(sharding_attrs)) { in ExtractInputsForLogicalDevices() local236 const auto& sharding_attr = sharding_attr_and_index.value(); in ExtractInputsForLogicalDevices()237 const auto input_index = sharding_attr_and_index.index(); in ExtractInputsForLogicalDevices()