Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Ddistributed.h42 std::shared_ptr<DistributedRuntimeClient> GetDistributedRuntimeClient(
Ddistributed.cc32 std::shared_ptr<DistributedRuntimeClient> GetDistributedRuntimeClient( in GetDistributedRuntimeClient() function
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc386 m.def("get_distributed_runtime_client", &GetDistributedRuntimeClient); in PYBIND11_MODULE()