Tensorflow SparseTensorDenseMatMul\U grad没有支持的GPU设备内核
在使用图中的tf.sparse\u tensor\u densite\u matmulop构建模型时，我遇到了一个问题。部分错误信息粘贴如下： 这是否意味着没有GPU内核支持来计算“sparsetensordensemular\u grad”的梯度？我可以在会话配置中使用“允许\u soft\u placement=true”成功构建模型。然而，由于某些特殊原因，我需要所有的计算都保持在GPU上。有人知道如何解决这个问题吗？或者我需要自己实现这个操作的CUDA内核？非常感谢Tensorflow SparseTensorDenseMatMul\U grad没有支持的GPU设备内核,tensorflow,Tensorflow,在使用图中的tf.sparse\u tensor\u densite\u matmulop构建模型时，我遇到了一个问题。部分错误信息粘贴如下： 这是否意味着没有GPU内核支持来计算“sparsetensordensemular\u grad”的梯度？我可以在会话配置中使用“允许\u soft\u placement=true”成功构建模型。然而，由于某些特殊原因，我需要所有的计算都保持在GPU上。有人知道如何解决这个问题吗？或者我需要自己实现这个操作的CUDA内核？非常感谢 tensorflow
tensorflow.python.framework.errors.InvalidArgumentError: Cannot assign a device to node 'gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1': Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available.
[[Node: gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1 = Slice[Index=DT_INT32, T=DT_INT64, _device="/device:GPU:0"](Placeholder_2, gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1/begin, gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1/size)]]
Caused by op u'gradients/softmax_linear/SparseTensorDenseMatMul/SparseTensorDenseMatMul_grad/Slice_1', defined at: