- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 46 for TensorHandle (0.16 sec)
-
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/immediate_execution_tensor_handle.h
#include "tensorflow/core/platform/status.h" namespace tensorflow { // Abstract interface to a TensorHandle. // // A TensorHandle is management class around a Tensor which may track additional // metadata and synchronization. // // This allows us to hide concrete implementations of TensorHandle from header // files. The interface lists the common functionality that must be provided by
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/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_experimental_test.cc
/*is_async=*/async, /*enable_streaming_enqueue=*/true, /*in_flight_nodes_limit=*/0); TFE_ContextSetExecutorForThread(ctx, executor); TFE_TensorHandle* m = TestMatrixTensorHandle(ctx); TFE_Op* matmul = MatMulOp(ctx, m, m); TFE_TensorHandle* retvals[2] = {nullptr, nullptr}; int num_retvals = 2; TFE_Execute(matmul, &retvals[0], &num_retvals, status); EXPECT_EQ(1, num_retvals);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
// Try to pass a ParallelTensor to TPUReplicatedInput std::array<TFE_TensorHandle*, 1> correct_components{value_one.get()}; TensorHandlePtr combined_value = CreatePerDeviceValues( context.get(), correct_components, device_name, status.get()); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); std::array<TFE_TensorHandle*, 1> incorrect_components{combined_value.get()};
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/c_api_unified_experimental_test.cc
// Build an abstract input tensor. TFE_Context* eager_ctx = TF_ExecutionContextGetTFEContext(ctx, status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); TFE_TensorHandle* t = TestScalarTensorHandle(eager_ctx, 2.0f); TF_AbstractTensor* at = TF_CreateAbstractTensorFromEagerTensor(t, status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.h
void AllocateLoggingDevice(const char* name, bool* arrived_flag, bool* executed_flag, TFE_CustomDevice** device, void** device_info); TFE_TensorHandle* UnpackTensorHandle(TFE_TensorHandle* logged_tensor_handle, TF_Status* status);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 1.6K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
// component device. Status SummarizeValue(std::string& summary); std::vector<TensorHandlePtr> release_tensors() { return std::move(tensors_); } std::vector<TFE_TensorHandle*> tensors() const { std::vector<TFE_TensorHandle*> result; result.reserve(tensors_.size()); for (const TensorHandlePtr& tensor : tensors_) { result.emplace_back(tensor.get()); } return result; }
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/BUILD
"//tensorflow/c:c_api", "//tensorflow/c:tf_status_helper", "//tensorflow/c/eager:c_api", "//tensorflow/c/eager:c_api_experimental", "//tensorflow/c/eager:tfe_tensorhandle_internal", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:optional", "@com_google_absl//absl/types:variant", ], ) cc_library( name = "parallel_device_lib",
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 01 20:19:06 GMT 2024 - 4.8K bytes - Viewed (0) -
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)