Searched refs:GetDefaultDeviceAssignment1D (Results 1 – 3 of 3) sorted by relevance
115 StatusOr<std::vector<ClientAndPtr<PjRtDevice>>> GetDefaultDeviceAssignment1D(
88 PyClient::GetDefaultDeviceAssignment1D(int num_replicas) { in GetDefaultDeviceAssignment1D() function in xla::PyClient
216 &PyClient::GetDefaultDeviceAssignment1D) in PYBIND11_MODULE()