在GPU上执行哪种张量流操作?

时间:2020-09-08 14:20:44

标签: tensorflow tensorflow2.0

我启用日志,该日志指示通过以下代码在哪些设备上执行tensorflow ops。

tf.debugging.set_log_device_placement(True) 

我得到的输出如下:

I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op LogicalNot in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op Assert in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op AssignVariableOp in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op Fill in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op VarHandleOp in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op VarIsInitializedOp in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op LogicalNot in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op Assert in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op AssignVariableOp in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op RandomUniform in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op Sub in device /job:localhost/replica:0/task:0/device:GPU:0
I tensorflow/core/common_runtime/eager/execute.cc:611] Executing op Mul in device /job:localhost/replica:0/task:0/device:GPU:0

我想知道tensorflow是否有一个文档描述在GPU上执行哪种类型的操作。

0 个答案:

没有答案
相关问题