Home
last modified time | relevance | path

Searched refs:AddressableDevices (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_executable.h51 std::vector<ClientAndPtr<PjRtDevice>> AddressableDevices() const;
Dpy_executable.cc61 std::vector<ClientAndPtr<PjRtDevice>> PyExecutable::AddressableDevices() const { in AddressableDevices() function in xla::PyExecutable
Dxla.cc352 .def("local_devices", &PyExecutable::AddressableDevices) in PYBIND11_MODULE()