Searched refs:kUnassignedDevice (Results 1 – 1 of 1) sorted by relevance
42 constexpr int64 kUnassignedDevice = -2; variable189 if (rhs.UsesDevice(kUnassignedDevice)) { in AssignLeafSharding()192 if (lhs->UsesDevice(kUnassignedDevice)) { in AssignLeafSharding()253 instruction->shape(), HloSharding::AssignDevice(kUnassignedDevice)); in ApplyShardingFromUsers()363 if (instruction->sharding().UsesDevice(kUnassignedDevice)) { in ApplyDomainSharding()