Searched defs:unique_device (Results 1 – 3 of 3) sorted by relevance
212 auto unique_device = tuple_element_sharding.UniqueDevice(); in UsedDevices() local219 auto unique_device = UniqueDevice(); in UsedDevices() local357 absl::optional<int64> unique_device; in UniqueDevice() local
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
1579 absl::optional<int64> unique_device = absl::nullopt; in CheckAndUpdateDeviceAssignmentsInWhileBody() local