- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for TFE_Execute (0.24 sec)
-
tensorflow/c/eager/c_api_test.cc
TFE_TensorHandle* dummy = nullptr; TFE_Execute(add_op_dummy, &dummy, &num_retvals, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_DeleteTensorHandle(dummy); TFE_DeleteOp(add_op_dummy); } } TFE_TensorHandle* retval = nullptr; TFE_Execute(add_op, &retval, &num_retvals, status); EXPECT_EQ(1, num_retvals);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/eager/c_api.h
// is done. // TODO(agarwal): change num_retvals to int from int*. TF_CAPI_EXPORT extern void TFE_Execute(TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals, TF_Status* status); // Add a function (serialized FunctionDef protocol buffer) to ctx so // that it can be invoked using TFE_Execute. TF_CAPI_EXPORT extern void TFE_ContextAddFunctionDef(
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Viewed (1) -
tensorflow/c/eager/c_api_cluster_test.cc
TFE_OpSetDevice(matmul, remote_device_name, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* retvals[1]; int num_retvals = 1; TFE_Execute(matmul, &retvals[0], &num_retvals, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); auto* retval_task0 = TFE_TensorHandleCopyToDevice(retvals[0], ctx, local_device_name, status);
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.h
// // Attributes are forwarded to executed operations unmodified. // // The returned optional has a value if and only if `status` evaluates to // TF_OK. Bad statuses are forwarded from underlying `TFE_Execute` calls, or // if sanity checks on dtypes/metadata fail. absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> Execute( TFE_Context* context, const std::vector<ParallelTensor*>& inputs,
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/c_api_distributed_test.cc
if (remote) { TFE_OpSetDevice(func, task1_name, status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); } TFE_TensorHandle* retvals[1] = {nullptr}; int num_retvals = 1; TFE_Execute(func, &retvals[0], &num_retvals, status); EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); ASSERT_EQ(1, num_retvals); TFE_DeleteOp(func); TFE_DeleteTensorHandle(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/c_api_experimental_test.cc
TFE_OpAddInput(op, h, status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); std::vector<TFE_TensorHandle*> result; result.push_back(nullptr); int num_retvals = 1; TFE_Execute(op, result.data(), &num_retvals, status); TFE_DeleteOp(op); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); ASSERT_EQ(num_retvals, 1);
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/c_api_test_util.cc
if (!device_name.empty()) { TFE_OpSetDevice(op, device_name.c_str(), status); } if (TF_GetCode(status) != TF_OK) return nullptr; TFE_TensorHandle* var_handle = nullptr; int num_retvals = 1; TFE_Execute(op, &var_handle, &num_retvals, status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_DeleteOp(op); if (TF_GetCode(status) != TF_OK) return nullptr; CHECK_EQ(1, num_retvals);
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/parallel_device/parallel_device_lib.cc
absl::optional<int64_t> step_id = absl::nullopt); // Block until the previous `StartExecute` operation has executed. Forwards // the status from `TFE_Execute` and returns outputs if the status is OK. std::vector<TensorHandlePtr> Join(TF_Status* status); // Block until all Ops finished running on the thread. void AsyncWait(TF_Status* status); private:
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.cc
TF_Status* status) { int ret = -1; status->status = tensorflow::unwrap(op)->OutputLength(output_name, &ret); return ret; } void TFE_Execute(TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals, TF_Status* status) { tensorflow::ImmediateExecutionOperation* unwrapped_op = tensorflow::unwrap(op); status->status =
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)