- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 96 for dltensor (0.23 sec)
-
tensorflow/c/eager/dlpack_test.cc
const DLTensor* dltensor_out = &dlm_out->dl_tensor; EXPECT_EQ(dltensor_out->device.device_type, dltensor_in->device.device_type); EXPECT_EQ(dltensor_out->device.device_id, dltensor_in->device.device_id); EXPECT_EQ(dltensor_out->ndim, dltensor_in->ndim); EXPECT_EQ(dltensor_out->dtype.code, dltensor_in->dtype.code); EXPECT_EQ(dltensor_out->dtype.bits, dltensor_in->dtype.bits);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.h
#ifndef TENSORFLOW_C_EAGER_DLPACK_H_ #define TENSORFLOW_C_EAGER_DLPACK_H_ #include "tensorflow/c/eager/c_api.h" namespace tensorflow { // PyCapsule name for DLPack Tensor const char* const kDlTensorCapsuleName = "dltensor"; // Converts eager tensor handle to DLPack (DLManagedTensor*), and return the // void* for further PyCapsule construction. TF_CAPI_EXPORT extern void* TFE_HandleToDLPack(TFE_TensorHandle* h,
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sat Mar 28 08:41:24 GMT 2020 - 1.6K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor; dlm_tensor->manager_ctx = tf_dlm_tensor_ctx; dlm_tensor->deleter = &DLManagedTensorDeleter; dlm_tensor->dl_tensor.device = tf_dlm_context; int ndim = tensor->dims(); dlm_tensor->dl_tensor.ndim = ndim; dlm_tensor->dl_tensor.data = tf_dlm_data; dlm_tensor->dl_tensor.dtype = tf_dlm_type;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/abstract_tensor_handle.h
// Returns the status of the tensor handle. If it is a tfrt::TensorHandle, // the tensor handle can be an error and return non-OK status. virtual tensorflow::Status TensorHandleStatus() const; // Returns tensor shape. If tensor has unknown rank, shape remains untouched. virtual tensorflow::Status Shape( tensorflow::PartialTensorShape* shape) const = 0; // Returns tensor (full) type.
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Mar 03 00:30:36 GMT 2023 - 3K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_tensor_handle.h
limitations under the License. ==============================================================================*/ #ifndef TENSORFLOW_C_EAGER_IMMEDIATE_EXECUTION_TENSOR_HANDLE_H_ #define TENSORFLOW_C_EAGER_IMMEDIATE_EXECUTION_TENSOR_HANDLE_H_ #include "tensorflow/c/eager/abstract_tensor_handle.h" #include "tensorflow/c/tensor_interface.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/status.h" namespace tensorflow {
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Mar 10 21:56:24 GMT 2023 - 4.3K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
#ifndef TENSORFLOW_C_EXPERIMENTAL_NEXT_PLUGGABLE_DEVICE_TENSOR_PJRT_BUFFER_UTIL_H_ #define 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);
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/eager/abstract_tensor_handle.cc
See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ #include "tensorflow/c/eager/abstract_tensor_handle.h" namespace tensorflow { std::string AbstractTensorHandle::DebugString() const { PartialTensorShape shape; Status s = Shape(&shape); std::string shape_string; if (!s.ok()) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
absl::StatusOr<PJRT_Buffer*> GetPjRtCBufferFromTensor(const Tensor* tensor) { tensorflow::AsyncValueTensor* av_tensor = tensorflow::AsyncValueTensor::FromTensor(tensor); if (av_tensor == nullptr || av_tensor->GetBuffer() == nullptr) { return absl::InternalError("Input tensor does not have PjRtBuffer."); } auto* c_api_buffer = dynamic_cast<xla::PjRtCApiBuffer*>(av_tensor->GetBuffer().get()); if (c_api_buffer == nullptr) {
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/tensor_pjrt_buffer_util_test.cc
tensorflow::AsyncValueTensor* av_tensor = tensorflow::AsyncValueTensor::FromTensor(&tensor); av_tensor->SetBuffer(std::move(buffer)); TF_ASSERT_OK_AND_ASSIGN(auto c_buffer, GetPjRtCBufferFromTensor(&tensor)); EXPECT_THAT(c_buffer, NotNull()); } TEST(TensorPjRtBufferUtilTest, SetPjRtCBufferToTensorNotAsyncValueTensor) { tensorflow::Tensor tensor(DT_FLOAT, {1});
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/eager/immediate_execution_tensor_handle.cc
See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ #include "tensorflow/c/eager/immediate_execution_tensor_handle.h" namespace tensorflow { std::string ImmediateExecutionTensorHandle::DebugString() const { PartialTensorShape shape; std::string shape_string; if (Shape(&shape).ok()) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 2.1K bytes - Viewed (0)