Searched defs:unique_device (Results 1 – 2 of 2) sorted by relevance
123 auto unique_device = tuple_element_sharding.UniqueDevice(); in UsedDevices() local130 auto unique_device = UniqueDevice(); in UsedDevices() local249 absl::optional<int64> unique_device; in UniqueDevice() local
308 optional<int64> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormTraining() local412 optional<int64> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormInference() local590 auto unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormGrad() local