Home
last modified time | relevance | path

Searched refs:LocalService (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlocal_service.h39 class LocalService : public Service {
42 static StatusOr<std::unique_ptr<LocalService>> NewService(
73 explicit LocalService(const ServiceOptions& options,
75 LocalService(const LocalService&) = delete;
76 void operator=(const LocalService&) = delete;
Dlocal_service.cc48 /* static */ StatusOr<std::unique_ptr<LocalService>> LocalService::NewService( in NewService()
63 std::unique_ptr<LocalService> service( in NewService()
64 new LocalService(options, std::move(backend))); in NewService()
68 LocalService::LocalService(const ServiceOptions& options, in LocalService() function in xla::LocalService
101 LocalService::CompileExecutables( in CompileExecutables()
190 StatusOr<int> LocalService::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal()
196 StatusOr<const ShapedBuffer*> LocalService::GlobalDataToShapedBuffer( in GlobalDataToShapedBuffer()
207 StatusOr<GlobalDataHandle> LocalService::RegisterReplicatedBuffers( in RegisterReplicatedBuffers()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient_library.h100 static LocalService* GetXlaService(se::Platform* platform);
123 std::unique_ptr<LocalService> service;
Dlocal_client.h131 explicit LocalClient(LocalService* service) in LocalClient()
213 LocalService* local_service_;
Dclient_library.cc114 LocalService::NewService(service_options)); in GetOrCreateLocalClient()
129 /* static */ LocalService* ClientLibrary::GetXlaService( in GetXlaService()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddumped_computation_to_operation_list.cc63 LocalService* local_service = in RealMain()
Ddumped_computation_to_text.cc38 LocalService* local_service = in RealMain()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dxla_hlo_profile_test.cc135 LocalService* service = ClientLibrary::GetXlaService(client->platform()); in ExecuteAndFetchProfile()