- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for client (0.15 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
#include "xla/pjrt/c/pjrt_c_api.h" #include "xla/pjrt/pjrt_c_api_client.h" #include "tensorflow/core/framework/tensor.h" namespace tensorflow { StatusOr<PJRT_Buffer*> GetPjRtCBufferFromTensor(const Tensor* tensor); absl::Status SetPjRtCBufferToTensor(PJRT_Buffer* c_buffer, xla::PjRtCApiClient* c_api_client, Tensor* tensor);
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 09 19:12:59 GMT 2023 - 1.5K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type) { TF_ASSIGN_OR_RETURN(tsl::StatusOr<xla::PjRtClient*> pjrt_client, tensorflow::GetPjRtClient(device_type)); auto* pjrt_c_api_client = dynamic_cast<xla::PjRtCApiClient*>(*pjrt_client); if (pjrt_c_api_client == nullptr) { return absl::InternalError(absl::StrCat("PjRtClient for ", device_type.type_string(),
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 3.7K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_distributed_manager.h
int keep_alive_secs) = 0; // Set up a multi-client distributed execution environment. Must be called // on all tasks in the cluster. This call internally coordinates with other // tasks to initialize the eager context and TF server for multi-client // execution. virtual Status EnableCollectiveOps(const ServerDef& server_def) = 0;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 2.9K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.h
int num_options); // Resets the PjRt client for a device. After this, `TF_GetPjRtCClient` will // returns an error for that device. TF_CAPI_EXPORT extern void TF_ResetPjRtCClient(const char* device_type, TF_Status* status); // Gets the `PJRT_Client*` stored in TF global ResourceManager. TF_CAPI_EXPORT extern PJRT_Client* TF_GetPjRtCClient(const char* device_type,
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Dec 20 20:01:06 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/BUILD
"@com_google_absl//absl/types:span", "@local_tsl//tsl/distributed_runtime/coordination:coordination_service_agent", "@local_xla//xla/pjrt:pjrt_c_api_client", "@local_xla//xla/pjrt:pjrt_client", "@local_xla//xla/pjrt/c:pjrt_c_api_hdrs", "@local_xla//xla/pjrt/c:pjrt_c_api_helpers", ], ) # Plugin should include this target to avoid linking the C API implementation.
Plain Text - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 3.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.h
// Stores a function representation that can be used for execution or for // setting functional attributes of other composite ops e.g. control flow. typedef struct TF_AbstractFunction TF_AbstractFunction; // This allows the client to swap the implementation of the tracing engine. // Any future call to TF_CreateFunction will use the implementation defined // here. void TF_SetTracingImplementation(const char* name, TF_Status*);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:00 GMT 2021 - 7K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
CHECK_OK(pjrt::SetPjrtApi(DEVICE_CPU, GetPjrtApi())); } auto pjrt_client = xla::GetCApiClient(DEVICE_CPU); CHECK_OK(pjrt_client.status()); auto c_api_client = down_cast<xla::PjRtCApiClient*>(pjrt_client->get()); std::vector<int32_t> data(1, 0); xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); auto buffer = c_api_client->pjrt_c_client()->client->BufferFromHostBuffer(
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.h
// of block_size. std::unique_ptr<ExpiringLRUCache<GcsFileStat>> stat_cache; GCSFile(google::cloud::storage::Client&& gcs_client); // This constructor is used for testing purpose only. GCSFile(google::cloud::storage::Client&& gcs_client, bool compose, uint64_t block_size, size_t max_bytes, uint64_t max_staleness, uint64_t stat_cache_max_age, size_t stat_cache_max_entries); } GCSFile;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Aug 31 04:37:41 GMT 2020 - 5.2K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/BUILD
"//tensorflow/c:logging", "//tensorflow/c:tf_status", "//tensorflow/c/experimental/filesystem:filesystem_interface", "@com_github_googlecloudplatform_google_cloud_cpp//:storage_client", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:variant", ], ) cc_library( name = "gcs_helper", srcs = ["gcs_helper.cc"], hdrs = ["gcs_helper.h"],
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 27 18:00:18 GMT 2024 - 3.4K bytes - Viewed (0) -
tensorflow/c/eager/abstract_operation.h
// // WARNING: The destructor of this class is marked as protected to disallow // clients from directly destroying this object since it may manage it's own // lifetime through ref counting. Thus this must be allocated on the heap and // clients MUST call Release() in order to destroy an instance of this class. virtual void Release() = 0;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jul 14 16:20:41 GMT 2021 - 6.8K bytes - Viewed (0)