| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 332 se::Platform* LocalClient::platform() const { in platform()
336 int LocalClient::device_count() const { in device_count()
340 bool LocalClient::device_ordinal_supported(int device_ordinal) const { in device_ordinal_supported()
344 int LocalClient::default_device_ordinal() const { in default_device_ordinal()
348 const Backend& LocalClient::backend() const { in backend()
352 Backend* LocalClient::mutable_backend() { in mutable_backend()
356 StatusOr<std::vector<std::unique_ptr<LocalExecutable>>> LocalClient::Compile( in Compile()
399 StatusOr<ScopedShapedBuffer> LocalClient::LiteralToShapedBuffer( in LiteralToShapedBuffer()
415 StatusOr<Literal> LocalClient::ShapedBufferToLiteral( in ShapedBufferToLiteral()
423 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 | 50 xla::LocalClient* client() const { return client_; } in client()
59 void Acquire(xla::LocalClient* client, int ordinal) { in Acquire()
64 xla::LocalClient* client_ = nullptr;
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tpu_client.cc | 51 TpuDeviceState(se::StreamExecutor* executor, LocalClient* client,
61 LocalClient* client, bool asynchronous) in TpuDeviceState()
78 PjRtTpuClient(LocalClient* client,
92 LocalClient* client, in PjRtTpuClient()
137 LocalClient* client, in GetTpuDevices()
203 TF_ASSIGN_OR_RETURN(LocalClient * client, in GetTpuClient()
|
| D | local_device_state.h | 91 LocalDeviceState(se::StreamExecutor* executor, LocalClient* client,
100 LocalClient* client() const { return client_; } in client()
175 LocalClient* const client_;
|
| 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 | gpu_device.cc | 62 StatusOr<LocalClient*> GetGpuXlaClient() { in GetGpuXlaClient()
76 LocalClient* xla_client, bool asynchronous) { in BuildLocalDeviceStates()
316 TF_ASSIGN_OR_RETURN(LocalClient * xla_client, GetGpuXlaClient()); in GetGpuClient()
|
| D | cpu_device.cc | 40 TF_ASSIGN_OR_RETURN(LocalClient * client, in GetCpuClient()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 59 xla::LocalClient* client,
73 xla::LocalClient* client() const { return client_; } in client()
110 xla::LocalClient* client_;
|
| D | xla_compilation_cache.h | 47 XlaCompilationCache(xla::LocalClient* client, DeviceType device_type);
91 xla::LocalClient* client() const { return client_; } in client()
142 xla::LocalClient* const client_;
|
| D | xla_launch_util.h | 135 XlaComputationLaunchContext(xla::LocalClient* client,
182 xla::LocalClient* client_;
|
| D | xla_device.h | 70 xla::LocalClient* client() const;
192 xla::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 | xla_tensor.h | 52 xla::LocalClient* client, int device_ordinal);
|
| D | xla_device.cc | 161 xla::LocalClient* XlaDevice::Metadata::client() const { in client()
239 xla::StatusOr<xla::LocalClient*> XlaDevice::GetOrCreateClient() const { in GetOrCreateClient()
294 TF_ASSIGN_OR_RETURN(xla::LocalClient * client, GetOrCreateClient()); in GetDeviceContextLocked()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 78 xla::LocalClient* client, xla::LocalExecutable* executable, in XlaExecutableClosure()
90 xla::LocalClient* client() const { return client_; } in client()
101 xla::LocalClient* client_;
170 xla::LocalClient** client, in CompileToLocalExecutable()
191 *client = static_cast<xla::LocalClient*>(cache->client()); in CompileToLocalExecutable()
224 xla::LocalClient* client; in Compute()
373 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 | 115 const HloSnapshot& module, LocalClient* client) { in CompileExecutable()
223 LocalClient* client, const Options& opts) { in ReplayComputation()
433 LocalClient* client = ClientLibrary::LocalClientOrDie(); in RealMain()
|
| D | dumped_computation_to_operation_list.cc | 62 LocalClient* client = ClientLibrary::LocalClientOrDie(); in RealMain()
|