Home
last modified time | relevance | path

Searched refs:LocalDevices (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_client.h107 std::vector<ClientAndPtr<PjRtDevice>> LocalDevices();
Dpy_client.cc48 std::vector<ClientAndPtr<PjRtDevice>> PyClient::LocalDevices() { in LocalDevices() function in xla::PyClient
Dxla.cc208 .def("local_devices", &PyClient::LocalDevices) in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_placer_test.cc26 TEST(VirtualPlacerTest, LocalDevices) { in TEST() argument