Searched defs:cusparse_handle (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_sparse.cc | 209 static inline Status Gtsv2Impl(SparseFn op, cusparseHandle_t cusparse_handle, in Gtsv2Impl()
245 cusparseHandle_t cusparse_handle, in Gtsv2BufferSizeExtImpl()
284 cusparseHandle_t cusparse_handle, in Gtsv2StridedBatchImpl()
310 SparseFn op, cusparseHandle_t cusparse_handle, int m, const Scalar* dl, in Gtsv2StridedBatchBufferSizeImpl()
393 SparseFnT op, OpKernelContext* context, cusparseHandle_t cusparse_handle, in CsrmmImpl()
472 SparseFnT op, OpKernelContext* context, cusparseHandle_t cusparse_handle, in CsrmvImpl()
513 cusparseHandle_t cusparse_handle, in CsrmvExImpl()
552 cusparseHandle_t cusparse_handle, in SpMVImpl()
620 SparseFnT op, OpKernelContext* context, cusparseHandle_t cusparse_handle, in CsrgeamImpl()
660 SparseFnT op, OpKernelContext* context, cusparseHandle_t cusparse_handle, in Csrgeam2Impl()
[all …]
|