| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | client_library.cc | 84 /* static */ StatusOr<LocalClient*> ClientLibrary::GetOrCreateLocalClient( in GetOrCreateLocalClient() function in xla::ClientLibrary
89 return GetOrCreateLocalClient(default_options); in GetOrCreateLocalClient()
92 /* static */ StatusOr<LocalClient*> ClientLibrary::GetOrCreateLocalClient( in GetOrCreateLocalClient() function in xla::ClientLibrary
126 auto client_status = GetOrCreateLocalClient(); in LocalClientOrDie()
|
| H A D | client_library.h | 86 static StatusOr<LocalClient*> GetOrCreateLocalClient(
89 static StatusOr<LocalClient*> GetOrCreateLocalClient(
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | xla_hlo_profile_test.cc | 198 ClientLibrary::GetOrCreateLocalClient(platform)); in XLA_TEST_F()
278 ClientLibrary::GetOrCreateLocalClient(platform)); in XLA_TEST_F()
|
| H A D | multiple_devices_on_host_test.cc | 65 xla::ClientLibrary::GetOrCreateLocalClient(client_options)); in TestWithDeviceCount()
|
| H A D | local_client_test_base.cc | 124 ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie()), in LocalClientTestBase()
|
| H A D | compute_constant_test.cc | 57 ClientLibrary::GetOrCreateLocalClient(platform); in ClientOrDie()
|
| H A D | client_library_test_base.cc | 46 ClientLibrary::GetOrCreateLocalClient(client_options); in GetOrCreateLocalClientOrDie()
|
| H A D | dynamic_ops_test.cc | 761 ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie(); in BM_DynamicSlice()
|
| H A D | local_client_execute_test.cc | 939 ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie(); in BM_LocalClientOverhead()
|
| H A D | cpu_gpu_fusion_test.cc | 837 ClientLibrary::GetOrCreateLocalClient(client_options).ValueOrDie(); in BM_ParallelFusion()
|
| H A D | while_test.cc | 1255 ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie(); in BM_WhileLoop()
|
| H A D | dot_operation_test.cc | 1822 ClientLibrary::GetOrCreateLocalClient(client_options).ValueOrDie(); in DOT_ReorderContracting()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | interpreter_device.cc | 42 ClientLibrary::GetOrCreateLocalClient(options)); in GetInterpreterClient()
|
| H A D | cpu_device.cc | 41 ClientLibrary::GetOrCreateLocalClient(options)); in GetCpuClient()
|
| H A D | tpu_client.cc | 289 ClientLibrary::GetOrCreateLocalClient(options)); in GetTpuClient()
|
| H A D | gpu_device.cc | 165 return ClientLibrary::GetOrCreateLocalClient(options); in GetGpuXlaClient()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_device.cc | 162 auto client = xla::ClientLibrary::GetOrCreateLocalClient(platform_); in client()
258 return xla::ClientLibrary::GetOrCreateLocalClient(options); in GetOrCreateClient()
|
| H A D | xla_platform_info.cc | 101 auto client = xla::ClientLibrary::GetOrCreateLocalClient(client_options); in BuildXlaCompilationCache()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_jit_compiled_cpu_function.cc | 91 xla::ClientLibrary::GetOrCreateLocalClient(platform)); in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | outfeed_receiver_test.cc | 87 ClientLibrary::GetOrCreateLocalClient(options)); in GetCpuClientWithNonLocalDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/tests/ |
| H A D | raw_api_test.cc | 366 xla::ClientLibrary::GetOrCreateLocalClient(platform).ValueOrDie(); in XlaCompiledProgramShape()
|