Searched defs:Dtype (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/ |
| H A D | CUDASolver.h | 16 #define CUDASOLVER_GETRF_ARGTYPES(Dtype) \ argument
20 void getrf(CUDASOLVER_GETRF_ARGTYPES(Dtype)) { in getrf()
33 #define CUDASOLVER_GETRS_ARGTYPES(Dtype) \ argument
37 void getrs(CUDASOLVER_GETRS_ARGTYPES(Dtype)) { in getrs()
49 #define CUDASOLVER_SYTRF_BUFFER_ARGTYPES(Dtype) \ argument
53 void sytrf_bufferSize(CUDASOLVER_SYTRF_BUFFER_ARGTYPES(Dtype)) { in sytrf_bufferSize()
68 #define CUDASOLVER_SYTRF_ARGTYPES(Dtype) \ argument
73 void sytrf(CUDASOLVER_SYTRF_ARGTYPES(Dtype)) { in sytrf()
104 #define CUDASOLVER_GESVD_ARGTYPES(Dtype, Vtype) \ argument
109 void gesvd(CUDASOLVER_GESVD_ARGTYPES(Dtype, Vtype)) { in gesvd()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDABlas.h | 42 #define CUDABLAS_GEMM_ARGTYPES(Dtype) \ argument
47 #define CUDABLAS_GEMM_ARGS(Dtype) transa, transb, m, n, k, alpha, a, lda, b, ldb, beta, c, ldc argument
50 inline void gemm(CUDABLAS_GEMM_ARGTYPES(Dtype)) { in gemm()
68 inline void gemm_internal(CUDABLAS_GEMM_ARGTYPES(Dtype)) { in gemm_internal()
146 #define CUDABLAS_BGEMM_ARGTYPES(Dtype) \ argument
152 #define CUDABLAS_BGEMM_ARGS(Dtype) \ argument
156 inline void bgemm(CUDABLAS_BGEMM_ARGTYPES(Dtype)) { in bgemm()
174 inline void bgemm_internal(CUDABLAS_BGEMM_ARGTYPES(Dtype)) { in bgemm_internal()
191 #define CUDABLAS_TRSM_ARGTYPES(Dtype) \ argument
197 inline void trsm(CUDABLAS_TRSM_ARGTYPES(Dtype)) { in trsm()
[all …]
|
| H A D | CUDABlas.cpp | 216 #define GEMM_CHECK_ARGVALUES(Dtype) \ argument
226 #define BGEMM_CHECK_ARGVALUES(Dtype) \ argument
323 inline void bgemm_internal_cublaslt(CUDABLAS_BGEMM_ARGTYPES(Dtype)) { in bgemm_internal_cublaslt()
458 inline void bgemm_internal_cublas(CUDABLAS_BGEMM_ARGTYPES(Dtype)) { in bgemm_internal_cublas()
785 inline void gemm_internal_cublaslt(CUDABLAS_GEMM_ARGTYPES(Dtype)) { in gemm_internal_cublaslt()
791 inline void gemm_internal_cublas(CUDABLAS_GEMM_ARGTYPES(Dtype)) { in gemm_internal_cublas()
1807 #define GEMV_CHECK_ARGVALUES(Dtype) \ argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | types.h | 16 class Dtype; variable
34 explicit Dtype(int8_t type) in Dtype() function
36 explicit Dtype(ScalarType type) : scalar_type_(type), lanes_(1) {} in Dtype() function
37 Dtype(int8_t type, int64_t lanes) in Dtype() function
39 Dtype(ScalarType type, int64_t lanes) : scalar_type_(type), lanes_(lanes) {} in Dtype() function
40 Dtype(Dtype type, int64_t lanes) in Dtype() function
|
| H A D | loopnest.h | 22 class Dtype; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/ |
| H A D | bfloat16.cc | 63 static int Dtype() { return npy_type; } in Dtype() function
665 static int Dtype() { return NPY_UBYTE; } in Dtype() function
671 static int Dtype() { return NPY_USHORT; } in Dtype() function
680 static int Dtype() { return NPY_UINT; } in Dtype() function
686 static int Dtype() { return NPY_ULONG; } in Dtype() function
692 static int Dtype() { return NPY_ULONGLONG; } in Dtype() function
698 static int Dtype() { return NPY_BYTE; } in Dtype() function
704 static int Dtype() { return NPY_SHORT; } in Dtype() function
710 static int Dtype() { return NPY_INT; } in Dtype() function
716 static int Dtype() { return NPY_LONG; } in Dtype() function
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | types.h | 44 using Dtype = at::ScalarType; variable
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_runtime.cc | 3445 TEST(StaticRuntime, Dtype) { in TEST() argument
|