| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 334 se::Platform* LocalClient::platform() const { in platform()
338 int LocalClient::device_count() const { in device_count()
342 bool LocalClient::device_ordinal_supported(int device_ordinal) const { in device_ordinal_supported()
346 int LocalClient::default_device_ordinal() const { in default_device_ordinal()
350 const Backend& LocalClient::backend() const { in backend()
354 Backend* LocalClient::mutable_backend() { in mutable_backend()
358 StatusOr<std::vector<std::unique_ptr<LocalExecutable>>> LocalClient::Compile( in Compile()
401 StatusOr<ScopedShapedBuffer> LocalClient::LiteralToShapedBuffer( in LiteralToShapedBuffer()
417 StatusOr<Literal> LocalClient::ShapedBufferToLiteral( in ShapedBufferToLiteral()
425 StatusOr<const ShapedBuffer*> LocalClient::GlobalDataToShapedBuffer( in GlobalDataToShapedBuffer()
[all …]
|
| D | local_client.h | 129 class LocalClient : public Client {
131 explicit LocalClient(LocalService* service) in LocalClient() function
134 LocalClient(const LocalClient&) = delete;
135 void operator=(const LocalClient&) = delete;
|
| D | client_library.h | 88 static StatusOr<LocalClient*> GetOrCreateLocalClient(
91 static StatusOr<LocalClient*> GetOrCreateLocalClient(
96 static LocalClient* LocalClientOrDie();
125 std::unique_ptr<LocalClient> client;
|
| D | client_library.cc | 82 /* 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/ |
| D | xrt_device.h | 53 xla::LocalClient* client() const { return client_; } in client()
63 void Acquire(xla::LocalClient* client, int ordinal,
66 xla::LocalClient* client_ = nullptr;
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | local_device_state.h | 91 LocalDeviceState(se::StreamExecutor* executor, LocalClient* client,
101 LocalClient* client() const { return client_; } in client()
175 LocalClient* const client_;
|
| D | tpu_client.cc | 53 TpuDeviceState(se::StreamExecutor* executor, LocalClient* client,
63 LocalClient* client, in TpuDeviceState()
83 LocalClient* client, in PjRtTpuClient()
195 LocalClient* client, in GetTpuDevices()
268 TF_ASSIGN_OR_RETURN(LocalClient * client, in GetTpuClient()
|
| D | tracked_device_buffer_test.cc | 31 LocalClient* client) { in MakeArray()
52 LocalClient* client = ClientLibrary::LocalClientOrDie(); in TEST()
96 LocalClient* client = ClientLibrary::LocalClientOrDie(); in TEST()
|
| D | interpreter_device.cc | 41 TF_ASSIGN_OR_RETURN(LocalClient * client, in GetInterpreterClient()
|
| D | cpu_device.cc | 40 TF_ASSIGN_OR_RETURN(LocalClient * client, in GetCpuClient()
|
| D | tpu_client.h | 57 PjRtTpuClient(LocalClient* client,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 60 xla::LocalClient* client,
74 xla::LocalClient* client() const { return client_; } in client()
111 xla::LocalClient* client_;
|
| D | xla_compilation_cache.h | 47 XlaCompilationCache(xla::LocalClient* client, DeviceType device_type);
105 xla::LocalClient* client() const { return client_; } in client()
162 xla::LocalClient* const client_;
|
| D | xla_launch_util.h | 142 XlaComputationLaunchContext(xla::LocalClient* client,
188 xla::LocalClient* client_;
|
| D | xla_device.h | 70 xla::LocalClient* client() const;
197 StatusOr<xla::LocalClient*> GetOrCreateClient() const;
|
| D | xla_compile_on_demand_op.cc | 49 xla::LocalClient* client = static_cast<xla::LocalClient*>(cache->client()); in Run()
|
| D | get_compiler_ir.cc | 44 xla::LocalClient* local_client, in BuildExecutable()
151 xla::LocalClient* local_client = cache->client(); in GetCompilerIr()
|
| D | xla_tensor.h | 52 xla::LocalClient* client, int device_ordinal);
|
| D | xla_device.cc | 161 xla::LocalClient* XlaDevice::Metadata::client() const { in client()
248 StatusOr<xla::LocalClient*> XlaDevice::GetOrCreateClient() const { in GetOrCreateClient()
303 TF_ASSIGN_OR_RETURN(xla::LocalClient * client, GetOrCreateClient()); in GetDeviceContextLocked()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 86 xla::LocalClient* client, xla::LocalExecutable* executable, in XlaExecutableClosure()
98 xla::LocalClient* client() const { return client_; } in client()
109 xla::LocalClient* client_;
179 bool may_alias_resource_update, xla::LocalClient** client, in CompileToLocalExecutable()
201 *client = static_cast<xla::LocalClient*>(cache->client()); in CompileToLocalExecutable()
232 xla::LocalClient* client; in Compute()
392 xla::LocalClient* client; in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute()
64 LocalClient* const client, in TestWithDeviceCount()
|
| D | xla_hlo_profile_test.cc | 131 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()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_test.cc | 73 xla::LocalClient* client = xla::ClientLibrary::LocalClientOrDie(); in TEST()
111 xla::LocalClient* client = xla::ClientLibrary::LocalClientOrDie(); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | sample_harness.cc | 37 xla::LocalClient* client(xla::ClientLibrary::LocalClientOrDie()); in main()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 113 const HloSnapshot& module, LocalClient* client, const Options& opts) { in CompileExecutable()
232 LocalClient* client, const Options& opts) { in ReplayComputation()
436 LocalClient* client = ClientLibrary::LocalClientOrDie(); in RealMain()
|