Searched defs:unique_device (Results 1 – 3 of 3) sorted by relevance
275 optional<int64> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormTraining() local362 optional<int64> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormInference() local539 auto unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormGrad() local
288 auto unique_device = tuple_element_sharding.UniqueDevice(); in UsedDevices() local295 auto unique_device = UniqueDevice(); in UsedDevices() local433 absl::optional<int64> unique_device; in UniqueDevice() local
1595 absl::optional<int64> unique_device = absl::nullopt; in CheckAndUpdateDeviceAssignmentsInWhileBody() local