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.h38 class LocalService : public Service {
41 static StatusOr<std::unique_ptr<LocalService>> NewService(
72 explicit LocalService(const ServiceOptions& options,
74 LocalService(const LocalService&) = delete;
75 void operator=(const LocalService&) = delete;
Dlocal_service.cc47 /* static */ StatusOr<std::unique_ptr<LocalService>> LocalService::NewService( in NewService()
62 std::unique_ptr<LocalService> service( in NewService()
63 new LocalService(options, std::move(backend))); in NewService()
67 LocalService::LocalService(const ServiceOptions& options, in LocalService() function in xla::LocalService
119 StatusOr<std::unique_ptr<Executable>> LocalService::CompileExecutable( in CompileExecutable()
183 StatusOr<int> LocalService::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal()
189 StatusOr<const ShapedBuffer*> LocalService::GlobalDataToShapedBuffer( in GlobalDataToShapedBuffer()
200 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.h108 explicit LocalClient(LocalService* service) in LocalClient()
189 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()