- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 46 for TensorHandle (0.2 sec)
-
tensorflow/c/eager/c_api_debug_test.cc
TFE_ContextOptions* opts = TFE_NewContextOptions(); TFE_Context* ctx = TFE_NewContext(opts, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_DeleteContextOptions(opts); TFE_TensorHandle* h = TestScalarTensorHandle(ctx, 1.0f); TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 06 22:10:09 GMT 2020 - 2.3K bytes - Viewed (0) -
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/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_unified_experimental_eager.cc
} return wrap(static_cast<AbstractContext*>(unwrap(c_ctx))); } TF_AbstractTensor* TF_CreateAbstractTensorFromEagerTensor(TFE_TensorHandle* t, TF_Status* s) { return wrap(static_cast<AbstractTensorHandle*>(unwrap(t))); } TFE_TensorHandle* TF_AbstractTensorGetEagerTensor(TF_AbstractTensor* at, TF_Status* s) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jun 25 04:40:46 GMT 2020 - 3.2K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
int num_retvals = 1; TFE_TensorHandle* var_value = nullptr; TFE_Execute(op.get(), &var_value, &num_retvals, status); if (TF_GetCode(status) != TF_OK) return nullptr; return TensorHandlePtr(var_value); } void Variable::GeneralAssignment(const char* op_name, TFE_Context* context, TFE_TensorHandle* value, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test.cc
TFE_DeleteContextOptions(opts); TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx); TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(ctx); const char remote_device_name[] = "/job:localhost/replica:0/task:1/device:CPU:0"; auto* h0_task1 =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 12 00:14:22 GMT 2020 - 5.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
const std::string& parallel_device_name, TFE_Context* context, std::unique_ptr<ParallelTensor> t, TF_Status* status) { // The resulting TensorHandle owns an opaque pointer to "device memory", which // for a ParallelDevice is really a ParallelTensor. When the TensorHandle is // deleted, it will call ParallelTensorDeallocator to free the struct. ParallelTensor* t_released = t.release();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_debug.cc
#include "tensorflow/c/eager/tfe_tensorhandle_internal.h" #include "tensorflow/c/tf_status_internal.h" #include "tensorflow/core/common_runtime/eager/tensor_handle.h" #include "tensorflow/core/platform/status.h" using tensorflow::string; namespace { std::vector<int64_t> TensorShapeAsVector(const tensorflow::TensorHandle& handle, tensorflow::Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 11 01:20:50 GMT 2021 - 2.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test_util.cc
TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status); EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx); TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(ctx); std::vector<TFE_TensorHandle*> handles_task0; if (heavy_load_on_streaming_rpc) { // Send 50 tensor copy requests to simulate that there have been some RPC
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Dec 11 22:56:03 GMT 2020 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.h
// void* for further PyCapsule construction. TF_CAPI_EXPORT extern void* TFE_HandleToDLPack(TFE_TensorHandle* h, TF_Status* status); // Converts DLPack (DLManagedTensor*) to eager tensor handle. TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_HandleFromDLPack(void* dlm, TF_Status* status,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sat Mar 28 08:41:24 GMT 2020 - 1.6K bytes - Viewed (0)