- Sort Score
- Result 10 results
- Languages All
Results 11 - 16 of 16 for TFE_Execute (0.29 sec)
-
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_testlib.h
} TFE_OpSetDevice(op.get(), device, 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); } template <typename value_type>
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
TFE_OpAddInput(op, t->tensor, s); } else { TFE_OpAddInput(op, input, s); } if (TF_GetCode(s) != TF_OK) return; } std::vector<TFE_TensorHandle*> op_outputs(*num_outputs); TFE_Execute(op, op_outputs.data(), num_outputs, s); TFE_DeleteOp(op); if (TF_GetCode(s) != TF_OK) return; std::vector<TFE_TensorHandle*> unwrapped_outputs; unwrapped_outputs.reserve(op_outputs.size());
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) -
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_remote_test_util.cc
// The input handles should never change since they have been mirrored. ASSERT_FALSE(remote_arg->HasLocalMirror(nullptr)); } TFE_TensorHandle* retvals[1]; int num_retvals = 1; TFE_Execute(matmul, &retvals[0], &num_retvals, status); EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); // TODO(gjn): Add support for waiting on async local mirrors if (!remote && !async && !remote_func_outputs) {
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/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)