- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for TensorHandle (0.22 sec)
-
tensorflow/c/eager/dlpack.cc
// Gets tensor from eager tensor handle. const Tensor* GetTensorFromHandle(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return nullptr; } tensorflow::TensorHandle* handle = tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h)); if (handle->Type() != TensorHandle::LOCAL) { status->status = tensorflow::errors::InvalidArgument(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
#include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_internal.h" #include "tensorflow/c/eager/tfe_context_internal.h" #include "tensorflow/c/eager/tfe_op_internal.h" #include "tensorflow/c/eager/tfe_tensorhandle_internal.h" #include "tensorflow/c/tf_buffer_internal.h" #include "tensorflow/compiler/jit/flags.h" #include "tensorflow/core/common_runtime/eager/attr_builder.h" #include "tensorflow/core/common_runtime/eager/context.h"
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/immediate_execution_tensor_handle.cc
} Status s; const char* device_name = DeviceName(&s); if (!s.ok()) { device_name = "<error fetching device name>"; } return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string, ", dtype=", DataType_Name(DataType()), ", device=\"", device_name, "\")"); } Status ImmediateExecutionTensorHandle::SummarizeValue(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 2.1K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
std::unique_ptr<ParallelTensor> ParallelDevice::CopyToParallelDevice( TFE_Context* context, TFE_TensorHandle* tensor, TF_Status* status) const { std::vector<TensorHandlePtr> components; components.reserve(underlying_devices_.size()); for (const std::string& underlying_device_name : underlying_devices_) { TFE_TensorHandle* t = TFE_TensorHandleCopyToDevice( tensor, context, underlying_device_name.c_str(), status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api_distributed_test.cc
const char task2_name[] = "/job:localhost/replica:0/task:2/device:CPU:0"; // Create one variable per task. TFE_TensorHandle* h0 = TestVariable(ctx, 1.0, task1_name); TFE_TensorHandle* h1 = TestVariable(ctx, 2.0, task2_name); TFE_TensorHandle* h2 = TestVariable(ctx, 3.0, task0_name); // Add a sync point to make sure that variables have been initialized // before the function execution starts.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
using tensorflow::string; using tensorflow::tstring; TFE_TensorHandle* TestScalarTensorHandle(TFE_Context* ctx, float value) { float data[] = {value}; TF_Status* status = TF_NewStatus(); TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, nullptr, 0, status); memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); TFE_TensorHandle* th = TFE_NewTensorHandleFromTensor(ctx, t, status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/eager/abstract_tensor_handle.cc
Status s = Shape(&shape); std::string shape_string; if (!s.ok()) { shape_string = "<error computing shape>"; } else { shape_string = shape.DebugString(); } return absl::StrCat("TensorHandle(shape=", shape_string, ", dtype=", DataType_Name(DataType()), ", type=", FullType().DebugString(), ")"); } Status AbstractTensorHandle::TensorHandleStatus() const {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
} TFE_TensorHandle* TFE_NewTensorHandle(const TF_Tensor* t, TF_Status* status) { tensorflow::Tensor tensor; status->status = tensorflow::TF_TensorToTensor(t, &tensor); if (!status->status.ok()) return nullptr; return tensorflow::wrap(tensorflow::TensorHandle::CreateLocalHandle(tensor)); } void TFE_DeleteTensorHandle(TFE_TensorHandle* h) { if (h == nullptr) return;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/c_api_experimental.cc
return new TF_Tensor{t}; } TFE_TensorHandle* TFE_NewTensorHandleFromTensor(TFE_Context* ctx, TF_Tensor* t, TF_Status* status) { return tensorflow::wrap( tensorflow::unwrap(ctx)->CreateLocalHandle(t->tensor)); } TFE_TensorHandle* TFE_CreatePackedTensorHandle(TFE_Context* ctx, TFE_TensorHandle** handles,
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) -
tensorflow/c/eager/unified_api_testutil.cc
return absl::OkStatus(); } Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus); TFE_TensorHandle* result_t = TF_AbstractTensorGetEagerTensor(wrap(t), status.get()); TF_RETURN_IF_ERROR(StatusFromTF_Status(status.get())); *result_tensor = TFE_TensorHandleResolve(result_t, status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0)