- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 38 for handlers (0.18 sec)
-
tensorflow/c/c_api.cc
if (!status->status.ok()) { delete lib_handle; return nullptr; } return lib_handle; } TF_Buffer TF_GetOpList(TF_Library* lib_handle) { return lib_handle->op_list; } void TF_DeleteLibraryHandle(TF_Library* lib_handle) { if (lib_handle == nullptr) return; tensorflow::port::Free(const_cast<void*>(lib_handle->op_list.data)); delete lib_handle; } TF_Buffer* TF_GetAllOpList() {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/dlpack_test.cc
dltensor_in->strides = strides.data(); TFE_TensorHandle* handle = TFE_HandleFromDLPack(&dlm_in, status, ctx); ASSERT_NE(handle, nullptr) << TF_Message(status) << " (shape=[" << absl::StrJoin(shape, ",") << "], strides=[" << absl::StrJoin(strides, ",") << "])"; auto* dlm_out = static_cast<DLManagedTensor*>(TFE_HandleToDLPack(handle, status)); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad.cc
#include "tensorflow/c/experimental/gradients/nn_grad.h" #include "absl/types/span.h" #include "tensorflow/c/eager/abstract_tensor_handle.h" #include "tensorflow/c/eager/immediate_execution_context.h" #include "tensorflow/c/eager/immediate_execution_tensor_handle.h" #include "tensorflow/c/experimental/ops/array_ops.h" #include "tensorflow/c/experimental/ops/math_ops.h" #include "tensorflow/c/experimental/ops/nn_ops.h"
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
TFE_TensorHandle* var_handle0 = TestVariable(ctx, 1.0, dev0_name); EXPECT_NE(var_handle0, nullptr); TFE_TensorHandle* var_handle1 = TestVariable(ctx, 2.0, dev1_name); EXPECT_NE(var_handle1, nullptr); TFE_TensorHandle* value_handle = nullptr; ReadVariable(ctx, var_handle1, &value_handle); CheckTFE_TensorHandleHasFloats(value_handle, {2}); TFE_DeleteTensorHandle(value_handle);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
/*expected_max_outputs=*/1, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); const std::vector<std::unique_ptr<ParallelTensor>>& handles = *outputs; std::vector<ParallelTensor*> handle_inputs; handle_inputs.reserve(handles.size()); for (auto& handle : handles) { handle_inputs.push_back(handle.get()); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/abstract_tensor_handle.cc
", dtype=", DataType_Name(DataType()), ", type=", FullType().DebugString(), ")"); } Status AbstractTensorHandle::TensorHandleStatus() const { // Tensor handles in current runtime don't carry error info and this method // should always return OK status. return absl::OkStatus(); }
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/parallel_device/parallel_device_testlib.cc
if (TF_GetCode(status) != TF_OK) return; TFE_TensorHandle* result_handles[num_replicas]; int num_retvals = num_replicas; TFE_Execute(op.get(), result_handles, &num_retvals, status); if (TF_GetCode(status) != TF_OK) return; for (int i = 0; i < num_replicas; ++i) { (*components)[i].reset(result_handles[i]); } } TensorHandlePtr Multiply(TFE_Context* context, TFE_TensorHandle* first,
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_distributed_test.cc
// Pack 3 variable handles into one TFE_TensorHandle. // When remote is false, function device is placed on task0. Handle types are // REMOTE, REMOTE, LOCAL on task0. When remote is true, function device is // placed on task1, Handle types are LOCAL, REMOTE, LOCAL on task1. int num_replicas = 3; std::vector<TFE_TensorHandle*> handles = {h0, h1, h2}; TFE_TensorHandle* packed_handle =
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/parallel_device/parallel_device_test.cc
TFE_OpAddInput(op.get(), input, status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_TensorHandle* result_handle; int num_retvals = 1; TFE_Execute(op.get(), &result_handle, &num_retvals, status); if (TF_GetCode(status) != TF_OK) return nullptr; return TensorHandlePtr(result_handle); } void TestCollective(bool async) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/custom_device_testutil.cc
TFE_CustomDeviceTensorHandleMethods handle_methods; handle_methods.num_dims = &LoggedTensorNumDims; handle_methods.dim = &LoggedTensorDim; handle_methods.deallocator = &LoggedTensorDeallocator; return TFE_NewCustomDeviceTensorHandle(context, logging_device_name.c_str(), dtype, t.release(), handle_methods, status); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0)