/external/tensorflow/tensorflow/compiler/xla/client/ |
D | local_client.cc | 316 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 …]
|
D | local_client.h | 106 class LocalClient : public Client { 108 explicit LocalClient(LocalService* service) in LocalClient() function 111 LocalClient(const LocalClient&) = delete; 112 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 | 46 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/ |
D | local_device_state.h | 45 LocalDeviceState(se::StreamExecutor* executor, LocalClient* client, 54 LocalClient* client() const { return client_; } in client() 121 LocalClient* const client_;
|
D | shared_device_buffer_test.cc | 28 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()
|
D | local_client.h | 126 std::string platform_name, LocalClient* client, 153 LocalClient* client() const { return client_; } in client() 170 LocalClient* client_;
|
D | local_device_state.cc | 28 LocalClient* client, in LocalDeviceState()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_device_context.h | 59 xla::LocalClient* client, 73 xla::LocalClient* client() const { return client_; } in client() 107 xla::LocalClient* client_;
|
D | xla_compilation_cache.h | 47 XlaCompilationCache(xla::LocalClient* client, DeviceType device_type); 89 xla::LocalClient* client() const { return client_; } in client() 140 xla::LocalClient* const client_;
|
D | xla_launch_util.h | 119 XlaComputationLaunchContext(xla::LocalClient* client, 164 xla::LocalClient* client_;
|
D | xla_device.h | 70 xla::LocalClient* client() const; 186 xla::StatusOr<xla::LocalClient*> GetOrCreateClient() const;
|
D | xla_tensor.h | 54 xla::LocalClient* client, int device_ordinal);
|
D | xla_device.cc | 160 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()
|
D | xla_tensor.cc | 48 xla::LocalClient* client, in AllocateShapedBuffer()
|
/external/tensorflow/tensorflow/compiler/jit/kernels/ |
D | xla_ops.cc | 107 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/ |
D | multiple_devices_on_host_test.cc | 37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute() 63 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()
|
D | local_client_test_base.h | 139 LocalClient* local_client_;
|
/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() 423 LocalClient* client = ClientLibrary::LocalClientOrDie(); in RealMain()
|
D | dumped_computation_to_operation_list.cc | 62 LocalClient* client = ClientLibrary::LocalClientOrDie(); in RealMain()
|
D | dumped_computation_to_text.cc | 37 LocalClient* client = ClientLibrary::LocalClientOrDie(); in RealMain()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | tf2xla_test.cc | 73 xla::LocalClient* client = xla::ClientLibrary::LocalClientOrDie(); in TEST()
|