HomeSort by relevance Sort by last modified time
    Searched refs:LocalClient (Results 1 - 21 of 21) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/client/
client_library.h 76 static StatusOr<LocalClient*> GetOrCreateLocalClient(
78 static StatusOr<LocalClient*> GetOrCreateLocalClient(
83 static LocalClient* LocalClientOrDie();
112 std::unique_ptr<LocalClient> client;
local_client.h 62 friend class LocalClient;
64 // Constructor invoked by LocalClient.
111 class LocalClient : public Client {
113 explicit LocalClient(LocalService* service)
116 LocalClient(const LocalClient&) = delete;
117 void operator=(const LocalClient&) = delete;
140 // TODO(b/69670845): Remove the 'Local' from the name when LocalClient does
147 // TODO(b/69670845): Remove the 'Local' from the name when LocalClient does
local_client.cc 225 se::Platform* LocalClient::platform() const {
229 int LocalClient::device_count() const {
233 bool LocalClient::device_ordinal_supported(int device_ordinal) const {
237 int LocalClient::default_device_ordinal() const {
241 const Backend& LocalClient::backend() const {
245 Backend* LocalClient::mutable_backend() {
249 StatusOr<std::unique_ptr<LocalExecutable>> LocalClient::Compile(
269 LocalClient::LiteralToShapedBuffer(const Literal& literal, int device_ordinal,
284 StatusOr<std::unique_ptr<Literal>> LocalClient::ShapedBufferToLiteral(
293 Status LocalClient::TransferToInfeedLocal(const Literal& literal
    [all...]
client_library.cc 71 /* static */ StatusOr<LocalClient*> ClientLibrary::GetOrCreateLocalClient(
78 /* static */ StatusOr<LocalClient*> ClientLibrary::GetOrCreateLocalClient(
103 instance->client = MakeUnique<LocalClient>(instance->service.get());
104 LocalClient* cl = instance->client.get();
111 /* static */ LocalClient* ClientLibrary::LocalClientOrDie() {
  /external/tensorflow/tensorflow/compiler/jit/
xla_compilation_cache.h 49 XlaCompilationCache(xla::LocalClient* client, DeviceType device_type);
72 xla::LocalClient* client() const { return client_; }
84 xla::LocalClient* const client_;
xla_device.h 57 xla::LocalClient* client() const;
98 xla::LocalClient* client() const;
xla_device.cc 154 xla::LocalClient* XlaDevice::Metadata::client() const {
190 xla::LocalClient* XlaDevice::client() const {
xla_compilation_cache.cc 40 XlaCompilationCache::XlaCompilationCache(xla::LocalClient* client,
  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc 30 LocalClient* g_local_client GUARDED_BY(g_local_client_mutex) = nullptr;
53 LocalClient* GetOrCreateLocalClient() {
68 LocalClient* client = GetOrCreateLocalClient();
76 LocalClient* client = GetOrCreateLocalClient();
86 LocalClient* client = GetOrCreateLocalClient();
102 LocalClient* client, int device_ordinal, const Literal& arg) {
111 LocalClient* client = GetOrCreateLocalClient();
124 LocalClient* client = GetOrCreateLocalClient();
135 LocalClient* client = GetOrCreateLocalClient();
234 LocalClient* client = GetOrCreateLocalClient()
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
tf2xla_test.cc 71 xla::LocalClient* client = xla::ClientLibrary::LocalClientOrDie();
xla_jit_compiled_cpu_function.cc 116 TF_ASSIGN_OR_RETURN(xla::LocalClient * client,
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
sample_harness.cc 37 xla::LocalClient* client(xla::ClientLibrary::LocalClientOrDie());
  /external/tensorflow/tensorflow/compiler/xla/tests/
local_client_test_base.h 74 // A base class for tests which exercise the LocalClient interface.
87 // LocalClient::LiteralToShapedBuffer.
134 LocalClient* local_client_;
xla_hlo_profile_test.cc 120 void ExecuteAndFetchProfile(string* profile_output, LocalClient* client,
185 TF_ASSERT_OK_AND_ASSIGN(LocalClient * client,
254 TF_ASSERT_OK_AND_ASSIGN(LocalClient * client,
dynamic_ops_test.cc 669 LocalClient* client =
local_client_execute_test.cc     [all...]
while_test.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/tools/
dumped_computation_to_operation_list.cc 65 LocalClient* client = ClientLibrary::LocalClientOrDie();
dumped_computation_to_text.cc 39 LocalClient* client = ClientLibrary::LocalClientOrDie();
  /external/tensorflow/tensorflow/compiler/jit/kernels/
xla_launch_op.cc 250 xla::LocalClient* client = static_cast<xla::LocalClient*>(cache->client());
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_cost_analysis_test.cc 61 static_cast<LocalClient*>(client_)->platform()))),

Completed in 646 milliseconds