Searched defs:sparse_prefix (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_sparse.cc | 194 #define SPARSE_FN(method, sparse_prefix) cusparse##sparse_prefix##method argument
195 #define SPARSE_NAME(method, sparse_prefix) "cusparse" #sparse_prefix #method argument
196 #define BUFSIZE_FN(method, sparse_prefix) \ argument
219 #define GTSV2_INSTANCE(Scalar, sparse_prefix) \ argument
231 #define GTSV2_NO_PIVOT_INSTANCE(Scalar, sparse_prefix) \ argument
256 #define GTSV2_BUFFER_SIZE_INSTANCE(Scalar, sparse_prefix) \ argument
269 #define GTSV2_NO_PIVOT_BUFFER_SIZE_INSTANCE(Scalar, sparse_prefix) \ argument
295 #define GTSV2_STRIDED_BATCH_INSTANCE(Scalar, sparse_prefix) \ argument
320 #define GTSV2_STRIDED_BATCH_BUFFER_SIZE_INSTANCE(Scalar, sparse_prefix) \ argument
413 #define CSRMM_INSTANCE(Scalar, sparse_prefix) \ argument
[all …]
|
| D | rocm_sparse.cc | 160 #define SPARSE_FN(method, sparse_prefix) wrap::hipsparse##sparse_prefix##method argument
195 #define CSRMM_INSTANCE(Scalar, sparse_prefix) \ argument
230 #define CSRMV_INSTANCE(Scalar, sparse_prefix) \ argument
281 #define CSRGEMM_INSTANCE(Scalar, sparse_prefix) \ argument
315 #define CSR2CSC_INSTANCE(Scalar, sparse_prefix) \ argument
|