Home
last modified time | relevance | path

Searched refs:computeType (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
DcublasLt_11_0.inc135 cublasComputeType_t computeType, cudaDataType_t scaleType) {
140 return func_ptr(matmulDesc, size, computeType, scaleType);
144 cublasLtMatmulDesc_t *matmulDesc, cublasComputeType_t computeType,
150 return func_ptr(matmulDesc, computeType, scaleType);
309 cublasLtHandle_t lightHandle, cublasComputeType_t computeType,
318 return func_ptr(lightHandle, computeType, scaleType, Atype, Btype, Ctype,
323 cublasLtHandle_t lightHandle, cublasComputeType_t computeType,
333 return func_ptr(lightHandle, computeType, scaleType, Atype, Btype, Ctype,
Dcusolver_dense_11_0.inc4607 cudaDataType computeType, size_t *workspaceInBytes) {
4613 return func_ptr(handle, params, uplo, n, dataTypeA, A, lda, computeType,
4620 void *A, int64_t lda, cudaDataType computeType, void *pBuffer,
4627 return func_ptr(handle, params, uplo, n, dataTypeA, A, lda, computeType,
4648 cudaDataType dataTypeTau, const void *tau, cudaDataType computeType,
4657 computeType, workspaceInBytes);
4663 cudaDataType dataTypeTau, void *tau, cudaDataType computeType,
4672 computeType, pBuffer, workspaceInBytes, info);
4678 cudaDataType computeType, size_t *workspaceInBytes) {
4684 return func_ptr(handle, params, m, n, dataTypeA, A, lda, computeType,
[all …]
Dcudnn_6_0.inc453 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
460 computeType);
471 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
478 computeType);
497 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
504 computeType);
510 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
517 dilationA, mode, computeType);
1777 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
1785 computeType);
[all …]
Dcusparse_11_0.inc6652 cusparseDnVecDescr_t vecY, const void *result, cudaDataType computeType,
6659 return func_ptr(handle, opX, vecX, vecY, result, computeType, bufferSize);
6665 cudaDataType computeType, void *externalBuffer) {
6671 return func_ptr(handle, opX, vecX, vecY, result, computeType, externalBuffer);
6740 cusparseDnVecDescr_t vecY, cudaDataType computeType, cusparseSpMVAlg_t alg,
6748 return func_ptr(handle, opA, alpha, matA, vecX, beta, vecY, computeType, alg,
6755 cusparseDnVecDescr_t vecY, cudaDataType computeType, cusparseSpMVAlg_t alg,
6763 return func_ptr(handle, opA, alpha, matA, vecX, beta, vecY, computeType, alg,
6788 cusparseDnVecDescr_t vecY, cudaDataType computeType, cusparseSpSVAlg_t alg,
6796 return func_ptr(handle, opA, alpha, matA, vecX, vecY, computeType, alg,
[all …]
Dcudnn_7_0.inc497 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
504 computeType);
515 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
522 computeType);
541 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
548 computeType);
554 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
561 dilationA, mode, computeType);
Dcudnn_7_3.inc497 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
504 computeType);
515 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
522 computeType);
541 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
548 computeType);
554 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
561 dilationA, mode, computeType);
Dcudnn_7_1.inc497 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
504 computeType);
515 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
522 computeType);
541 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
548 computeType);
554 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
561 dilationA, mode, computeType);
Dcudnn_7_4.inc497 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
504 computeType);
515 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
522 computeType);
541 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
548 computeType);
554 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
561 dilationA, mode, computeType);
Dcudnn_7_6.inc662 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
669 computeType);
680 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
687 computeType);
706 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
713 computeType);
719 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
726 dilationA, mode, computeType);
Dcudnn_8_0.inc1103 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
1110 computeType);
1121 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
1128 computeType);
1134 cudnnConvolutionMode_t mode, cudnnDataType_t computeType) {
1141 computeType);
1147 cudnnConvolutionMode_t *mode, cudnnDataType_t *computeType) {
1154 dilationA, mode, computeType);
Dcusparse_10_1.inc8146 void *result, cudaDataType computeType, void *externalBuffer) {
8152 return func_ptr(handle, opX, vecX, vecY, result, computeType, externalBuffer);
8158 const void *result, cudaDataType computeType, size_t *bufferSize) {
8164 return func_ptr(handle, opX, vecX, vecY, result, computeType, bufferSize);
8170 const void *beta, const cusparseDnVecDescr_t vecY, cudaDataType computeType,
8178 return func_ptr(handle, opA, alpha, matA, vecX, beta, vecY, computeType, alg,
8185 const void *beta, const cusparseDnVecDescr_t vecY, cudaDataType computeType,
8193 return func_ptr(handle, opA, alpha, matA, vecX, beta, vecY, computeType, alg,
8201 cusparseDnMatDescr_t matC, cudaDataType computeType, cusparseSpMMAlg_t alg,
8209 return func_ptr(handle, opA, opB, alpha, matA, matB, beta, matC, computeType,
[all …]
Dcusparse_10_2.inc8146 void *result, cudaDataType computeType, void *externalBuffer) {
8152 return func_ptr(handle, opX, vecX, vecY, result, computeType, externalBuffer);
8158 const void *result, cudaDataType computeType, size_t *bufferSize) {
8164 return func_ptr(handle, opX, vecX, vecY, result, computeType, bufferSize);
8170 const void *beta, const cusparseDnVecDescr_t vecY, cudaDataType computeType,
8178 return func_ptr(handle, opA, alpha, matA, vecX, beta, vecY, computeType, alg,
8185 const void *beta, const cusparseDnVecDescr_t vecY, cudaDataType computeType,
8193 return func_ptr(handle, opA, alpha, matA, vecX, beta, vecY, computeType, alg,
8201 cusparseDnMatDescr_t matC, cudaDataType computeType, cusparseSpMMAlg_t alg,
8209 return func_ptr(handle, opA, opB, alpha, matA, matB, beta, matC, computeType,
[all …]
Dcublas_10_0.inc1980 void *C, cudaDataType Ctype, int ldc, cudaDataType computeType,
1990 Btype, ldb, beta, C, Ctype, ldc, computeType, algo);
2670 cudaDataType computeType, cublasGemmAlgo_t algo) {
2680 computeType, algo);
2691 cudaDataType computeType, cublasGemmAlgo_t algo) {
2701 batchCount, computeType, algo);
Dcublas_11_0.inc2279 void *C, cudaDataType Ctype, int ldc, cublasComputeType_t computeType,
2289 Btype, ldb, beta, C, Ctype, ldc, computeType, algo);
2969 cublasComputeType_t computeType, cublasGemmAlgo_t algo) {
2979 computeType, algo);
2990 cublasComputeType_t computeType, cublasGemmAlgo_t algo) {
3000 batchCount, computeType, algo);
Dcublas_10_2.inc2105 void *C, cudaDataType Ctype, int ldc, cudaDataType computeType,
2115 Btype, ldb, beta, C, Ctype, ldc, computeType, algo);
2795 cudaDataType computeType, cublasGemmAlgo_t algo) {
2805 computeType, algo);
2816 cudaDataType computeType, cublasGemmAlgo_t algo) {
2826 batchCount, computeType, algo);
Dcublas_10_1.inc2105 void *C, cudaDataType Ctype, int ldc, cudaDataType computeType,
2115 Btype, ldb, beta, C, Ctype, ldc, computeType, algo);
2795 cudaDataType computeType, cublasGemmAlgo_t algo) {
2805 computeType, algo);
2816 cudaDataType computeType, cublasGemmAlgo_t algo) {
2826 batchCount, computeType, algo);
Dcublas_9_0.inc1955 void *C, cudaDataType Ctype, int ldc, cudaDataType computeType,
1965 Btype, ldb, beta, C, Ctype, ldc, computeType, algo);
/external/pytorch/aten/src/ATen/native/sparse/cuda/
DSparseMatMul.cu267 cudaDataType computeType = at::cuda::getCudaDataType<scalar_t>(); in operator ()() local
272 …TORCH_CHECK(prop->major >= 5 && !((10*prop->major + prop->minor) < 53 && computeType == CUDA_R_16F… in operator ()()
274 TORCH_CHECK(!(prop->major < 8 && computeType == CUDA_R_16BF), in operator ()()
287 computeType, in operator ()()
308 computeType, in operator ()()
324 computeType, in operator ()()
343 computeType, in operator ()()
375 computeType, in operator ()()
/external/pytorch/aten/src/ATen/cuda/
DCUDABlas.cpp325 cublasComputeType_t computeType = CUBLAS_COMPUTE_32F; in bgemm_internal_cublaslt() local
329 computeType = CUBLAS_COMPUTE_64F; in bgemm_internal_cublaslt()
334 computeType = CUBLAS_COMPUTE_32F_FAST_TF32; in bgemm_internal_cublaslt()
339 computeType = CUBLAS_COMPUTE_64F; in bgemm_internal_cublaslt()
358 CuBlasLtMatmulDescriptor computeDesc(computeType, scaleType); in bgemm_internal_cublaslt()
451 computeType, in bgemm_internal_cublaslt()
1201 cublasComputeType_t computeType = CUBLAS_COMPUTE_32F; in gemm_and_bias() local
1205 computeType = CUBLAS_COMPUTE_64F; in gemm_and_bias()
1210 computeType = CUBLAS_COMPUTE_32F_FAST_TF32; in gemm_and_bias()
1220 CuBlasLtMatmulDescriptor computeDesc(computeType, scaleType); in gemm_and_bias()
[all …]
/external/skia/include/core/
DSkRRect.h506 void computeType();
/external/skia/src/core/
DSkRRect.cpp279 this->computeType(); in scaleRadii()
382 void SkRRect::computeType() { in computeType() function in SkRRect
928 intersection.computeType(); in ConservativeIntersect()
/external/pytorch/aten/src/ATen/native/cuda/linalg/
DCUDASolver.h426 …int64_t lda, cudaDataType computeType, size_t *workspaceInBytesOnDevice, size_t *workspaceInBytesO…
430 …int64_t lda, cudaDataType computeType, void *bufferOnDevice, size_t workspaceInBytesOnDevice, void…
DCUDASolver.cpp1044 …int64_t lda, cudaDataType computeType, size_t *workspaceInBytesOnDevice, size_t *workspaceInBytesO… in xpotrf_buffersize() argument
1046 …handle, params, uplo, n, dataTypeA, A, lda, computeType, workspaceInBytesOnDevice, workspaceInByte… in xpotrf_buffersize()
1052 …int64_t lda, cudaDataType computeType, void *bufferOnDevice, size_t workspaceInBytesOnDevice, void… in xpotrf() argument
1055 …handle, params, uplo, n, dataTypeA, A, lda, computeType, bufferOnDevice, workspaceInBytesOnDevice,… in xpotrf()