张量流中的稀疏矩阵乘法是否使用GPU?(cuSparse?)

时间:2018-02-06 16:27:19

标签: tensorflow sparse-matrix matrix-multiplication

我正在尝试使用tensorflow进行稀疏矩阵乘法。

但是,当我看到sparse_matmul_op.cc文件时,它似乎没有启动任何cuda内核,例如cuBlas或cuSparse。

是在CPU上计算的吗?不是GPU?

0 个答案:

没有答案