Searched refs:sharding_attrs (Results 1 – 1 of 1) sorted by relevance
221 auto sharding_attrs = in ExtractInputsForLogicalDevices() local226 if (!sharding_attrs) { in ExtractInputsForLogicalDevices()235 for (const auto& sharding_attr_and_index : llvm::enumerate(sharding_attrs)) { in ExtractInputsForLogicalDevices()