Home
last modified time | relevance | path

Searched refs:LocalClient (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc316 se::Platform* LocalClient::platform() const { in platform()
320 int LocalClient::device_count() const { in device_count()
324 bool LocalClient::device_ordinal_supported(int device_ordinal) const { in device_ordinal_supported()
328 int LocalClient::default_device_ordinal() const { in default_device_ordinal()
332 const Backend& LocalClient::backend() const { in backend()
336 Backend* LocalClient::mutable_backend() { in mutable_backend()
340 StatusOr<std::vector<std::unique_ptr<LocalExecutable>>> LocalClient::Compile( in Compile()
366 StatusOr<ScopedShapedBuffer> LocalClient::LiteralToShapedBuffer( in LiteralToShapedBuffer()
382 StatusOr<Literal> LocalClient::ShapedBufferToLiteral( in ShapedBufferToLiteral()
390 StatusOr<const ShapedBuffer*> LocalClient::GlobalDataToShapedBuffer( in GlobalDataToShapedBuffer()
[all …]
Dlocal_client.h106 class LocalClient : public Client {
108 explicit LocalClient(LocalService* service) in LocalClient() function
111 LocalClient(const LocalClient&) = delete;
112 void operator=(const LocalClient&) = delete;
Dclient_library.h88 static StatusOr<LocalClient*> GetOrCreateLocalClient(
91 static StatusOr<LocalClient*> GetOrCreateLocalClient(
96 static LocalClient* LocalClientOrDie();
125 std::unique_ptr<LocalClient> client;
Dclient_library.cc82 /* static */ StatusOr<LocalClient*> ClientLibrary::GetOrCreateLocalClient( in GetOrCreateLocalClient()
90 /* static */ StatusOr<LocalClient*> ClientLibrary::GetOrCreateLocalClient( in GetOrCreateLocalClient()
115 instance->client = absl::make_unique<LocalClient>(instance->service.get()); in GetOrCreateLocalClient()
116 LocalClient* cl = instance->client.get(); in GetOrCreateLocalClient()
123 /* static */ LocalClient* ClientLibrary::LocalClientOrDie() { in LocalClientOrDie()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_device.h46 xla::LocalClient* client() const { return client_; } in client()
55 void Acquire(xla::LocalClient* client, int ordinal) { in Acquire()
60 xla::LocalClient* client_ = nullptr;
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_device_state.h45 LocalDeviceState(se::StreamExecutor* executor, LocalClient* client,
54 LocalClient* client() const { return client_; } in client()
121 LocalClient* const client_;
Dshared_device_buffer_test.cc28 LocalClient* client = ClientLibrary::LocalClientOrDie(); in TEST()
43 LocalClient* client = ClientLibrary::LocalClientOrDie(); in TEST()
71 LocalClient* client = ClientLibrary::LocalClientOrDie(); in TEST()
133 LocalClient* client = ClientLibrary::LocalClientOrDie(); in TEST()
Dlocal_client.h126 std::string platform_name, LocalClient* client,
153 LocalClient* client() const { return client_; } in client()
170 LocalClient* client_;
Dlocal_device_state.cc28 LocalClient* client, in LocalDeviceState()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.h59 xla::LocalClient* client,
73 xla::LocalClient* client() const { return client_; } in client()
107 xla::LocalClient* client_;
Dxla_compilation_cache.h47 XlaCompilationCache(xla::LocalClient* client, DeviceType device_type);
89 xla::LocalClient* client() const { return client_; } in client()
140 xla::LocalClient* const client_;
Dxla_launch_util.h119 XlaComputationLaunchContext(xla::LocalClient* client,
164 xla::LocalClient* client_;
Dxla_device.h70 xla::LocalClient* client() const;
186 xla::StatusOr<xla::LocalClient*> GetOrCreateClient() const;
Dxla_tensor.h54 xla::LocalClient* client, int device_ordinal);
Dxla_device.cc160 xla::LocalClient* XlaDevice::Metadata::client() const { in client()
238 xla::StatusOr<xla::LocalClient*> XlaDevice::GetOrCreateClient() const { in GetOrCreateClient()
293 TF_ASSIGN_OR_RETURN(xla::LocalClient * client, GetOrCreateClient()); in GetDeviceContextLocked()
Dxla_tensor.cc48 xla::LocalClient* client, in AllocateShapedBuffer()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc107 xla::LocalClient* client, xla::LocalExecutable* executable, in XlaExecutableClosure()
120 xla::LocalClient* client() const { return client_; } in client()
131 xla::LocalClient* client_;
275 absl::Span<const int> constants, bool lazy, xla::LocalClient** client, in CompileToLocalExecutable()
298 *client = static_cast<xla::LocalClient*>(cache->client()); in CompileToLocalExecutable()
346 xla::LocalClient* client; in Compute()
483 xla::LocalClient* client; in Compute()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute()
63 LocalClient* const client, in TestWithDeviceCount()
Dxla_hlo_profile_test.cc131 void ExecuteAndFetchProfile(string* profile_output, LocalClient* client, in ExecuteAndFetchProfile()
197 TF_ASSERT_OK_AND_ASSIGN(LocalClient * client, in XLA_TEST_F()
276 TF_ASSERT_OK_AND_ASSIGN(LocalClient * client, in XLA_TEST_F()
Dlocal_client_test_base.h139 LocalClient* local_client_;
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc37 xla::LocalClient* client(xla::ClientLibrary::LocalClientOrDie()); in main()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc115 const HloSnapshot& module, LocalClient* client) { in CompileExecutable()
223 LocalClient* client, const Options& opts) { in ReplayComputation()
423 LocalClient* client = ClientLibrary::LocalClientOrDie(); in RealMain()
Ddumped_computation_to_operation_list.cc62 LocalClient* client = ClientLibrary::LocalClientOrDie(); in RealMain()
Ddumped_computation_to_text.cc37 LocalClient* client = ClientLibrary::LocalClientOrDie(); in RealMain()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_test.cc73 xla::LocalClient* client = xla::ClientLibrary::LocalClientOrDie(); in TEST()

12