我启用日志,该日志指示通过以下代码在哪些设备上执行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上执行哪种类型的操作。