具有GPU支持的TensorFlow操作
是否有一种方法（或列表？）来确定所有提供GPU支持的tensorflow操作？ 现在，这对我来说是一个反复试验的过程——我尝试在GPU上放置一组操作。如果有效的话，甜心。如果没有，试试其他地方具有GPU支持的TensorFlow操作,tensorflow,gpu,Tensorflow,Gpu,是否有一种方法（或列表？）来确定所有提供GPU支持的tensorflow操作？ 现在，这对我来说是一个反复试验的过程——我尝试在GPU上放置一组操作。如果有效的话，甜心。如果没有，试试其他地方 这是迄今为止我发现的唯一相关（但没有帮助）的东西：不幸的是，似乎没有内置的方法来获得确切的列表，甚至无法检查特定的op。如上述问题所述，该调度是在本地C++代码中完成的：如果相应的内核已注册到代码> DEVICEXGPU ，则可以将特定操作分配给GPU。 我认为最简单的方法是grep“REGISTER\u
这是迄今为止我发现的唯一相关（但没有帮助）的东西：不幸的是，似乎没有内置的方法来获得确切的列表，甚至无法检查特定的op。如上述问题所述，该调度是在本地C++代码中完成的：如果相应的内核已注册到代码> DEVICEXGPU ，则可以将特定操作分配给GPU。 我认为最简单的方法是
grep“REGISTER\u KERNEL\u BUILDER”-r tensorflowREGISTER\u KERNEL\u BUILDERAdd