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