Searched defs:unique_device (Results 1 – 6 of 6) sorted by relevance
276 optional<int64_t> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormTraining() local365 optional<int64_t> unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormInference() local544 auto unique_device = batch_norm->sharding_unique_device(); in HandleBatchNormGrad() local
377 auto unique_device = tuple_element_sharding.UniqueDevice(); in UsedDevices() local384 auto unique_device = UniqueDevice(); in UsedDevices() local522 std::optional<int64_t> unique_device; in UniqueDevice() local
869 std::optional<int64_t> unique_device = std::nullopt; in CheckAndUpdateDeviceAssignmentsInWhileBody() local
94 torch::lazy::Unique<torch::lazy::BackendDevice> unique_device; in GetTensorsGraphInfo() local
577 Unique<BackendDevice> unique_device; in CollectSyncTensors() local
76 using unique_device = std::unique_ptr<libusb_device, DeviceDeleter>; typedef