Home
last modified time | relevance | path

Searched refs:GetOrCreateLocalClient (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dclient_library.cc82 /* static */ StatusOr<LocalClient*> ClientLibrary::GetOrCreateLocalClient( in GetOrCreateLocalClient() function in xla::ClientLibrary
87 return GetOrCreateLocalClient(default_options); in GetOrCreateLocalClient()
90 /* static */ StatusOr<LocalClient*> ClientLibrary::GetOrCreateLocalClient( in GetOrCreateLocalClient() function in xla::ClientLibrary
124 auto client_status = GetOrCreateLocalClient(); in LocalClientOrDie()
Dclient_library.h88 static StatusOr<LocalClient*> GetOrCreateLocalClient(
91 static StatusOr<LocalClient*> GetOrCreateLocalClient(
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc64 xla::ClientLibrary::GetOrCreateLocalClient(client_options)); in TestWithDeviceCount()
Dxla_hlo_profile_test.cc198 ClientLibrary::GetOrCreateLocalClient(platform)); in XLA_TEST_F()
277 ClientLibrary::GetOrCreateLocalClient(platform)); in XLA_TEST_F()
Dlocal_client_test_base.cc122 ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie()), in LocalClientTestBase()
Dcompute_constant_test.cc58 ClientLibrary::GetOrCreateLocalClient(platform); in ClientOrDie()
Dclient_library_test_base.cc47 ClientLibrary::GetOrCreateLocalClient(client_options); in GetOrCreateLocalClientOrDie()
Ddynamic_ops_test.cc741 ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie(); in BM_DynamicSlice()
Dlocal_client_execute_test.cc905 ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie(); in BM_LocalClientOverhead()
Dfusion_test.cc840 ClientLibrary::GetOrCreateLocalClient(client_options).ValueOrDie(); in BM_ParallelFusion()
Dwhile_test.cc1270 ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie(); in BM_WhileLoop()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc156 auto client = xla::ClientLibrary::GetOrCreateLocalClient(platform_); in client()
236 return xla::ClientLibrary::GetOrCreateLocalClient(platform_, allowed_devices_) in client()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_jit_compiled_cpu_function.cc84 xla::ClientLibrary::GetOrCreateLocalClient()); in Compile()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc239 auto client = xla::ClientLibrary::GetOrCreateLocalClient(client_options); in BuildCompilationCache()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc75 ClientLibrary::GetOrCreateLocalClient(options)); in Get()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc230 xla::ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie(); in XlaCompiledProgramShape()