- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for DeviceType (0.17 sec)
-
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
xla::PjRtCApiClient* c_api_client, Tensor* tensor); absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type); absl::Status ResetPjRtClient(const DeviceType& device_type); } // namespace tensorflow
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_test.cc
c_buffer, down_cast<xla::PjRtCApiClient*>(pjrt_client.get()), &tensor)); } TEST(TensorPjRtBufferUtilTest, GetPjRtCApiClientNotFound) { EXPECT_THAT( GetPjRtCApiClient(tensorflow::DeviceType(DEVICE_CPU)), StatusIs(error::NOT_FOUND, HasSubstr(absl::StrCat("PjRt client not found for device type ", DEVICE_CPU)))); }
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/next_pluggable_device/tensor_pjrt_buffer_util.cc
} else { av_tensor->SetBuffer(std::move(buffer)); } return absl::OkStatus(); } 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);
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/experimental/next_pluggable_device/c_api.cc
} absl::Status s = tensorflow::SetPjRtClientInTFGlobalResourceManager( tensorflow::DeviceType(device_type), std::move(*pjrt_client)); status->status = s; } void TF_ResetPjRtCClient(const char* device_type, TF_Status* status) { status->status = tensorflow::ResetPjRtClient(tensorflow::DeviceType(device_type)); } PJRT_Client* TF_GetPjRtCClient(const char* device_type, TF_Status* status) {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/eager/immediate_execution_tensor_handle.h
// Returns the device where the tensor was placed. virtual const char* BackingDeviceName(Status* status) const = 0; // Returns the device type which created the handle. virtual const char* DeviceType(Status* status) const = 0; // Returns the device ID which created the handle. virtual int DeviceId(Status* status) const = 0; // Returns a tensor for the handle. If tensor is remote, it will be copied.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Mar 10 21:56:24 GMT 2023 - 4.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
const char* device_type, TF_Status* status) { status->status = tensorflow::FindKernelDef( tensorflow::DeviceType(device_type), builder->BuildNodeDef(), /* def = */ nullptr, /* kernel_class_name = */ nullptr); } const char* TF_GetNumberAttrForOpListInput(const char* op_name, int input_index,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.cc
if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return nullptr; } return tensorflow::unwrap(h)->DeviceType(&status->status); } int TFE_TensorHandleDeviceID(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return -1; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3)