草庐IT

c++ - 了解 TensorFlow 中的操作注册和内核链接

coder 2023-11-16 原文

我是 TensorFlow 的新手，现在正在研究自定义操作开发。我已经阅读了官方教程，但我觉得幕后发生了很多事情，我并不总是想将我的自定义操作放在 user_ops 目录中。

因此，我占用了一个 example word2vec

它使用自定义的“Skipgram”操作，其注册在此处定义:
/word2vec_o​​ps.cc
其内核实现在这里:
/word2vec_kernels.cc

查看构建文件，我尝试构建单个目标

1) bazel build -c opt tensorflow/models/embedding:word2vec_o​​ps
这会按预期生成一堆目标文件。

2) bazel build -c opt tensorflow/models/embedding:word2vec_kernels
这点也一样。

3) bazel build -c opt tensorflow/models/embedding:word2vec_kernels:gen_word2vec

最后一次构建使用自定义规则，即 tf_op_gen_wrapper_py https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tensorflow.bzl#L197-L231

有趣的是，这仅取决于 Op Registration 而不是内核本身。

毕竟，如果我构建 py_binary 本身使用

bazel build -c opt tensorflow/models/embedding:word2vec

它工作正常，但我看不到内核 C++ 代码在哪里以及如何链接？

此外，我还想了解 tf_op_gen_wrapper_py 规则以及操作注册幕后的整个编译/链接过程。

谢谢。

最佳答案

当 adding a new kind of operation to TensorFlow ，主要有两个步骤:

  1. Registering the "op" ，其中涉及为操作定义接口(interface)，和

  2. Registering one or more "kernels" ，这涉及为操作定义实现，可能具有针对不同数据类型或设备类型(如 CPU 或 GPU)的专门实现。

这两个步骤都涉及编写 C++ 代码。 注册操作使用 REGISTER_OP() macro , 并注册内核使用 REGISTER_KERNEL_BUILDER() macro .这些宏创建静态初始值设定项，当加载包含它们的模块时运行。操作和内核注册有两种主要机制:

  1. 静态链接到核心 TensorFlow 库和静态初始化。

  2. 运行时动态链接，使用 tf.load_op_library()功能。

对于 "Skipgram"，我们使用选项 1(静态链接)。操作链接到核心 TensorFlow 库 here ，内核链接在 here 中. (请注意，这并不理想:word2vec 操作是在我们拥有 tf.load_op_library() 之前创建的，因此没有动态链接它们的机制。)因此操作和内核在您首次加载 TensorFlow 时注册(在 import tensorflow as tf 中)。如果它们是今天创建的，它们将被动态加载，这样它们只会在需要时才被注册。 (SyntaxNet 代码有一个 example 的动态加载。)

tf_op_gen_wrapper_py() rule在 Bazel 中获取 op 库依赖项列表，并为这些操作生成 Python 包装器。该规则仅取决于操作注册的原因是 Python 包装器完全由操作的接口(interface)决定，该接口(interface)在操作注册中定义。值得注意的是，Python 接口(interface)不知道是否有针对特定类型或设备的专用内核。包装器生成器将操作注册链接到 simple C++ binary 中。为每个注册的操作生成 Python 代码。 请注意，如果您使用 tf.load_op_library()，则无需自己调用包装器生成器，因为 tf.load_op_library() 会在运行时生成必要的代码.

关于c++ - 了解 TensorFlow 中的操作注册和内核链接，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/37548662/

TensorFlowampcodenoreferrerc++kernelregistrationswig

有关c++ - 了解 TensorFlow 中的操作注册和内核链接的更多相关文章

  1. ruby - 如何从 ruby​​ 中的字符串运行任意对象方法？ - 2

    总的来说，我对ruby​​还比较陌生，我正在为我正在创建的对象编写一些rspec测试用例。许多测试用例都非常基础，我只是想确保正确填充和返回值。我想知道是否有办法使用循环结构来执行此操作。不必为我要测试的每个方法都设置一个assertEquals。例如:describeitem,"TestingtheItem"doit"willhaveanullvaluetostart"doitem=Item.new#HereIcoulddotheitem.name.shouldbe_nil#thenIcoulddoitem.category.shouldbe_nilendend但我想要一些方法来使用

  2. ruby - 其他文件中的 Rake 任务 - 2

    我试图在一个项目中使用rake，如果我把所有东西都放到Rakefile中，它会很大并且很难读取/找到东西，所以我试着将每个命名空间放在lib/rake中它自己的文件中，我添加了这个到我的rake文件的顶部:Dir['#{File.dirname(__FILE__)}/lib/rake/*.rake'].map{|f|requiref}它加载文件没问题，但没有任务。我现在只有一个.rake文件作为测试，名为“servers.rake”，它看起来像这样:namespace:serverdotask:testdoputs"test"endend所以当我运行rakeserver:testid时

  3. ruby-on-rails - Ruby net/ldap 模块中的内存泄漏 - 2

    作为我的Rails应用程序的一部分，我编写了一个小导入程序，它从我们的LDAP系统中吸取数据并将其塞入一个用户表中。不幸的是，与LDAP相关的代码在遍历我们的32K用户时泄漏了大量内存，我一直无法弄清楚如何解决这个问题。这个问题似乎在某种程度上与LDAP库有关，因为当我删除对LDAP内容的调用时，内存使用情况会很好地稳定下来。此外，不断增加的对象是Net::BER::BerIdentifiedString和Net::BER::BerIdentifiedArray，它们都是LDAP库的一部分。当我运行导入时，内存使用量最终达到超过1GB的峰值。如果问题存在，我需要找到一些方法来更正我的代

  4. ruby-on-rails - Rails 3 中的多个路由文件 - 2

    Rails2.3可以选择随时使用RouteSet#add_configuration_file添加更多路由。是否可以在Rails3项目中做同样的事情？ 最佳答案 在config/application.rb中:config.paths.config.routes在Rails3.2(也可能是Rails3.1)中，使用:config.paths["config/routes"] 关于ruby-on-rails-Rails3中的多个路由文件，我们在StackOverflow上找到一个类似的问题

  5. ruby-on-rails - Rails - 一个 View 中的多个模型 - 2

    我需要从一个View访问多个模型。以前，我的links_controller仅用于提供以不同方式排序的链接资源。现在我想包括一个部分(我假设)显示按分数排序的顶级用户(@users=User.all.sort_by(&:score))我知道我可以将此代码插入每个链接操作并从View访问它，但这似乎不是“ruby方式”，我将需要在不久的将来访问更多模型。这可能会变得很脏，是否有针对这种情况的任何技术？注意事项:我认为我的应用程序正朝着单一格式和动态页面内容的方向发展，本质上是一个典型的网络应用程序。我知道before_filter但考虑到我希望应用程序进入的方向，这似乎很麻烦。最终从任何

  6. ruby-on-rails - Rails 3.2.1 中 ActionMailer 中的未定义方法 'default_content_type=' - 2

    我在我的项目中添加了一个系统来重置用户密码并通过电子邮件将密码发送给他，以防他忘记密码。昨天它运行良好(当我实现它时)。当我今天尝试启动服务器时，出现以下错误。=>BootingWEBrick=>Rails3.2.1applicationstartingindevelopmentonhttp://0.0.0.0:3000=>Callwith-dtodetach=>Ctrl-CtoshutdownserverExiting/Users/vinayshenoy/.rvm/gems/ruby-1.9.3-p0/gems/actionmailer-3.2.1/lib/action_mailer

  7. ruby-on-rails - 如何优雅地重启 thin + nginx？ - 2

    我的瘦服务器配置了nginx，我的ROR应用程序正在它们上运行。在我发布代码更新时运行thinrestart会给我的应用程序带来一些停机时间。我试图弄清楚如何优雅地重启正在运行的Thin实例，但找不到好的解决方案。有没有人能做到这一点？ 最佳答案 #Restartjustthethinserverdescribedbythatconfigsudothin-C/etc/thin/mysite.ymlrestartNginx将继续运行并代理请求。如果您将Nginx设置为使用多个上游服务器，例如server{listen80;server

  8. ruby-on-rails - Rails 应用程序中的 Rails : How are you using application_controller. rb 是新手吗？ - 2

    刚入门rails，开始慢慢理解。有人可以解释或给我一些关于在application_controller中编码的好处或时间和原因的想法吗？有哪些用例。您如何为Rails应用程序使用应用程序Controller？我不想在那里放太多代码，因为据我了解，每个请求都会调用此Controller。这是真的？ 最佳答案 ApplicationController实际上是您应用程序中的每个其他Controller都将从中继承的类(尽管这不是强制性的)。我同意不要用太多代码弄乱它并保持干净整洁的态度，尽管在某些情况下ApplicationContr

  9. ruby-on-rails - form_for 中不在模型中的自定义字段 - 2

    我想向我的Controller传递一个参数，它是一个简单的复选框，但我不知道如何在模型的form_for中引入它，这是我的观点:{:id=>'go_finance'}do|f|%>Transferirde:para:Entrada:"input",:placeholder=>"Quantofoiganho?"%>Saída:"output",:placeholder=>"Quantofoigasto?"%>Nota:我想做一个额外的复选框，但我该怎么做，模型中没有一个对象，而是一个要检查的对象，以便在Controller中创建一个ifelse，如果没有检查，请帮助我，非常感谢,谢谢

  10. ruby - rspec 需要 .rspec 文件中的 spec_helper - 2

    我注意到像bundler这样的项目在每个specfile中执行requirespec_helper我还注意到rspec使用选项--require，它允许您在引导rspec时要求一个文件。您还可以将其添加到.rspec文件中，因此只要您运行不带参数的rspec就会添加它。使用上述方法有什么缺点可以解释为什么像bundler这样的项目选择在每个规范文件中都需要spec_helper吗？ 最佳答案 我不在Bundler上工作，所以我不能直接谈论他们的做法。并非所有项目都checkin.rspec文件。原因是这个文件，通常按照当前的惯例，只

随机推荐