- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for Op (0.13 sec)
-
tensorflow/c/c_api.cc
delete lib_handle; } TF_Buffer* TF_GetAllOpList() { std::vector<tensorflow::OpDef> op_defs; tensorflow::OpRegistry::Global()->GetRegisteredOps(&op_defs); tensorflow::OpList op_list; for (const auto& op : op_defs) { *(op_list.add_op()) = op; } TF_Buffer* ret = TF_NewBuffer(); TF_CHECK_OK(MessageToBuffer(op_list, ret)); return ret; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
NodeDef node_def; tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op); node_def.set_name(op->Name()); node_def.set_op(op->Name()); for (int i = 0; i < num_inputs; ++i) { node_def.add_input("dummy_input"); } OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr()); const tensorflow::OpRegistrationData* op_reg_data; status->status =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
std::vector<const Node*>* body_nodes) TF_EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { if (num_opers == -1) { for (const Node* node : fn_body->graph.op_nodes()) { const auto& iter = input_nodes.find(node); if (iter == input_nodes.end()) { // This node is not referenced in inputs. Add it to the body. body_nodes->push_back(node); } else {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/eager/c_api_experimental.cc
const std::vector<std::string>& op_names = tensorflow::unwrap(ctx)->GetLoggedOpsTestonly(); std::ostringstream op_names_oss; for (const auto& op : op_names) { op_names_oss << op << ", "; } const std::string& op_names_str = op_names_oss.str(); void* data = tensorflow::port::Malloc(op_names_str.length()); op_names_str.copy(static_cast<char*>(data), op_names_str.length(), 0); buf->data = data;
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) -
tensorflow/c/c_api_test.cc
TF_DeleteStatus(status); ASSERT_EQ(TF_OK, code) << status_msg; // Test op list. TF_Buffer op_list_buf = TF_GetOpList(lib); tensorflow::OpList op_list; EXPECT_TRUE(op_list.ParseFromArray(op_list_buf.data, op_list_buf.length)); ASSERT_EQ(op_list.op_size(), 1); EXPECT_EQ("TestCApi1", op_list.op(0).name()); TF_DeleteLibraryHandle(lib); } #endif // !defined(TENSORFLOW_NO_SHARED_OBJECTS)
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3)