- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for TF_FinishOperation (0.14 sec)
-
tensorflow/c/c_api_test.cc
TF_SetAttrIntList(desc, "v", nullptr, 0); auto oper = TF_FinishOperation(desc, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); EXPECT_TF_META("v", 0, TF_ATTR_INT, -1); } TEST_F(CApiAttributesTest, Names) { auto desc = init("string"); TF_SetAttrString(desc, "v", "bunny", 5); auto oper = TF_FinishOperation(desc, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 97K bytes - Viewed (0) -
tensorflow/c/c_api.h
TF_Graph* graph, const char* op_type, const char* oper_name); // Operation will only be added to *graph when TF_FinishOperation() is // called (assuming TF_FinishOperation() does not return an error). // *graph must not be deleted until after TF_FinishOperation() is // called. TF_CAPI_EXPORT extern TF_OperationDescription* TF_NewOperation( TF_Graph* graph, const char* op_type, const char* oper_name);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
TF_Status* status = TF_NewStatus(); TF_Operation* arg = TF_FinishOperation(arg_descr, status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); TF_OperationDescription* id_descr = TF_NewOperation(function_graph, "Identity", "id"); TF_SetAttrType(id_descr, "T", TF_INT32); TF_AddInput(id_descr, {arg, 0}); TF_Operation* id = TF_FinishOperation(id_descr, status);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 03 03:14:26 UTC 2023 - 31.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
TF_OperationDescription* placeholder_desc = TF_NewOperation(body.get(), "Placeholder", "Placeholder"); TF_SetAttrType(placeholder_desc, "dtype", TF_FLOAT); TF_Operation* placeholder_op = TF_FinishOperation(placeholder_desc, status); if (TF_GetCode(status) != TF_OK) return; TF_Output x{placeholder_op, 0}; TF_OperationDescription* reduce_desc =
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Tue Aug 06 23:56:17 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
auto* tf_opdesc = op_.release(); if (tf_opdesc == nullptr) { return errors::InvalidArgument("AbstractOp is incomplete."); } TF_Status* s = TF_NewStatus(); auto* operation = TF_FinishOperation(tf_opdesc, s); TF_RETURN_IF_ERROR(StatusFromTF_Status(s)); TF_DeleteStatus(s); *num_retvals = TF_OperationNumOutputs(operation); for (int i = 0; i < *num_retvals; ++i) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/c/c_api_function_test.cc
TF_AddInput(desc, input); } // Set device to CPU because some ops inside the function might not be // available on GPU. TF_SetDevice(desc, "/cpu:0"); *op = TF_FinishOperation(desc, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); ASSERT_NE(*op, nullptr); } FunctionDef fdef() { tensorflow::FunctionDef fdef; EXPECT_TRUE(GetFunctionDef(func_, &fdef));
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Jul 20 22:08:54 UTC 2023 - 63.6K bytes - Viewed (0) -
tensorflow/c/c_api.cc
} else if (ret != nullptr) { desc->graph->graph.RemoveNode(ret); ret = nullptr; } } delete desc; return ToOperation(ret); } TF_Operation* TF_FinishOperation(TF_OperationDescription* desc, TF_Status* status) { mutex_lock l(desc->graph->mu); return TF_FinishOperationLocked(desc, status); } // TF_Operation functions
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 102.3K bytes - Viewed (0)