Searched refs:gpusparseDnMatDescr_t (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_sparse.h | 39 using gpusparseDnMatDescr_t = cusparseDnMatDescr_t; variable
292 const gpusparseDnMatDescr_t matB, const Scalar* beta,
293 gpusparseDnMatDescr_t matC, gpusparseSpMMAlg_t alg,
305 const gpusparseDnMatDescr_t matB, const Scalar* beta,
306 gpusparseDnMatDescr_t matC, gpusparseSpMMAlg_t alg,
|
| D | cuda_sparse.cc | 438 const gpusparseDnMatDescr_t matB, const Scalar* beta, \
439 gpusparseDnMatDescr_t matC, cusparseSpMMAlg_t alg, size_t* bufferSize) \
455 const gpusparseDnMatDescr_t matB, const Scalar* beta, \
456 gpusparseDnMatDescr_t matC, cusparseSpMMAlg_t alg, int8* buffer) const { \
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 836 gpusparseDnMatDescr_t matB, matC; in Compute()
|