- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 20 for TFE_DeleteOp (0.27 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
"/job:localhost/replica:0/task:0/device:CPU:1"}; ParallelDevice parallel_device(std::move(devices)); std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op( TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
TFE_DeleteTensorHandle(input2); TFE_DeleteTensorHandle(identity_ret[0]); TFE_DeleteTensorHandle(identity_ret[1]); TFE_DeleteOp(cloned_identity); TFE_DeleteOp(original_identity); TFE_DeleteOp(original_var_op); TFE_DeleteOp(cloned); TF_DeleteStatus(status); TFE_DeleteContext(ctx); } TEST(CAPI, ShareVariableAcrossContextsWorks) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/eager/custom_device_test.cc
ASSERT_TRUE(arrived); ASSERT_FALSE(executed); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( MatMulOp(context, hcpu, hdevice), TFE_DeleteOp); TFE_OpSetDevice(matmul.get(), name, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); TFE_TensorHandle* retval; int num_retvals = 1;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
const int64_t* dims, const int num_dims, const char* device, TF_Status* status) { std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( TFE_NewOp(context, "VarHandleOp", status), TFE_DeleteOp); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrType(op.get(), "dtype", type); TFE_OpSetAttrShape(op.get(), "shape", dims, num_dims, status);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/tfe_op_internal.h
// Wraps a pointer to an operation implementation. // // WARNING: Since the underlying object could be ref-counted a user of this // interface cannot destruct the underlying operation object. Instead, call // TFE_DeleteOp who calls Release() on the operation pointer and deletes // the TFE_Op structure. typedef struct TFE_Op TFE_Op; namespace tensorflow { DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionOperation, TFE_Op);
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Jun 17 19:39:13 GMT 2020 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
CheckTFE_TensorHandleHasFloats(retval_task0, {7, 10, 15, 22}); TFE_DeleteTensorHandle(retval_task0); TFE_DeleteTensorHandle(h0_task0); TFE_DeleteTensorHandle(retvals[0]); TFE_DeleteOp(matmul); TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); TFE_ExecutorWaitForAllPendingNodes(executor, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 23 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_testlib.h
TFE_Context* context, const std::array<TFE_TensorHandle*, num_replicas>& components, const char* device, TF_Status* status) { std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( TFE_NewOp(context, "TPUReplicatedInput", status), TFE_DeleteOp); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_OpSetAttrInt(op.get(), "N", num_replicas); for (int i = 0; i < num_replicas; ++i) {
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
<< TF_Message(status.get()); } { // Try to pass a non-parallel tensor to TPUReplicatedOutput std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( TFE_NewOp(context.get(), "TPUReplicatedOutput", status.get()), TFE_DeleteOp); if (TF_GetCode(status.get()) != TF_OK) return; TFE_OpSetAttrInt(op.get(), "num_replicas", 1); TFE_OpAddInput(op.get(), value_one.get(), status.get());
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/c_api_experimental_test.cc
/*input_shapes*/ {unknown_shape(), unknown_shape()}, /*input_tensors*/ {}, /*expected_shape*/ make_shape({kUnknownDim, kUnknownDim})); TFE_DeleteOp(matmul_op); // TODO(bgogul): Add some death tests where status is not OK. } TEST_F(ShapeInferenceTest, InfersShapesFromInputTensors) { // Prepare some tensors for shape.
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Jan 17 22:27:52 GMT 2023 - 13.1K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental_test.cc
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); TF_Tensor* r = TFE_TensorHandleResolve(result[0], status);
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1)