Searched defs:strideA (Results 1 – 4 of 4) sorted by relevance
| /external/eigen/Eigen/src/Core/arch/AltiVec/ |
| D | MatrixProductMMA.h | 252 Index strideA, in gemm_unrolled_MMA_iteration()
282 …st Scalar* blockB, Index rows, Index depth, Index cols, Scalar alpha, Index strideA, Index strideB… in gemmMMA()
498 Index strideA, in gemm_complex_unrolled_MMA_iteration()
541 …sScalar* blockBc, Index rows, Index depth, Index cols, Scalarc alpha, Index strideA, Index strideB… in gemm_complexMMA()
|
| /external/pytorch/aten/src/ATen/miopen/ |
| D | Descriptors.cpp | 73 int strideA[MIOPEN_DIM_MAX]; in operator <<() local
|
| /external/pytorch/aten/src/ATen/cudnn/ |
| D | Descriptors.cpp | 103 int strideA[CUDNN_DIM_MAX]; in operator <<() local
|
| /external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| D | CUDASolver.cpp | 477 …t handle, cusolverEigMode_t jobz, int rank, int m, int n, float *A, int lda, long long int strideA, in gesvdaStridedBatched_buffersize()
488 … handle, cusolverEigMode_t jobz, int rank, int m, int n, double *A, int lda, long long int strideA, in gesvdaStridedBatched_buffersize()
499 …lverEigMode_t jobz, int rank, int m, int n, c10::complex<float> *A, int lda, long long int strideA, in gesvdaStridedBatched_buffersize()
517 …verEigMode_t jobz, int rank, int m, int n, c10::complex<double> *A, int lda, long long int strideA, in gesvdaStridedBatched_buffersize()
536 …t handle, cusolverEigMode_t jobz, int rank, int m, int n, float *A, int lda, long long int strideA, in gesvdaStridedBatched()
547 … handle, cusolverEigMode_t jobz, int rank, int m, int n, double *A, int lda, long long int strideA, in gesvdaStridedBatched()
558 …lverEigMode_t jobz, int rank, int m, int n, c10::complex<float> *A, int lda, long long int strideA, in gesvdaStridedBatched()
578 …verEigMode_t jobz, int rank, int m, int n, c10::complex<double> *A, int lda, long long int strideA, in gesvdaStridedBatched()
|