Home
last modified time | relevance | path

Searched defs:unique_device (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc276 optional<int64_t> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormTraining() local
365 optional<int64_t> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormInference() local
544 auto unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormGrad() local
Dhlo_sharding.cc377 auto unique_device = tuple_element_sharding.UniqueDevice(); in UsedDevices() local
384 auto unique_device = UniqueDevice(); in UsedDevices() local
522 std::optional<int64_t> unique_device; in UniqueDevice() local
Dsharding_propagation.cc869 std::optional<int64_t> unique_device = std::nullopt; in CheckAndUpdateDeviceAssignmentsInWhileBody() local