Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
H A Ddistributed.cc35 std::shared_ptr<DistributedRuntimeClient> GetDistributedRuntimeClient( in GetDistributedRuntimeClient() function
42 return GetDistributedRuntimeClient(channel, options, in GetDistributedRuntimeClient()
H A Ddistributed.h43 std::shared_ptr<DistributedRuntimeClient> GetDistributedRuntimeClient(
H A Dclient.h132 std::unique_ptr<DistributedRuntimeClient> GetDistributedRuntimeClient(
H A Dclient.cc554 std::unique_ptr<DistributedRuntimeClient> GetDistributedRuntimeClient( in GetDistributedRuntimeClient() function
H A Dclient_server_test.cc61 return GetDistributedRuntimeClient(channel, client_options,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla.cc550 return GetDistributedRuntimeClient(address, options, in PYBIND11_MODULE()