Home
last modified time | relevance | path

Searched refs:type_prefix (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc322 #define DN_SOLVER_FN(method, type_prefix) cusolverDn##type_prefix##method argument
323 #define DN_SOLVER_NAME(method, type_prefix) "cusolverDn" #type_prefix #method argument
324 #define DN_BUFSIZE_FN(method, type_prefix) \ argument
325 cusolverDn##type_prefix##method##_bufferSize
328 #define BLAS_SOLVER_FN(method, type_prefix) cublas##type_prefix##method argument
329 #define BLAS_SOLVER_NAME(method, type_prefix) "cublas" #type_prefix #method argument
364 #define GEAM_INSTANCE(Scalar, type_prefix) \ argument
372 return GeamImpl(BLAS_SOLVER_FN(geam, type_prefix), cublas_handle_, transa, \
400 #define POTRF_INSTANCE(Scalar, type_prefix) \ argument
404 return PotrfImpl(DN_BUFSIZE_FN(potrf, type_prefix), \
[all …]
Drocm_solvers.cc180 #define BLAS_SOLVER_FN(method, type_prefix) \ argument
181 wrap::rocblas##_##type_prefix##method
227 #define TRSM_INSTANCE(Scalar, type_prefix) \ argument
236 return TrsmImpl(gpu_executor, BLAS_SOLVER_FN(trsm, type_prefix), \
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_context.cc134 #define DN_SOLVER_FN(method, type_prefix) cusolverDn##type_prefix##method argument
171 #define POTRF_INSTANCE(T, type_prefix) \ argument
176 return CusolverStatusToStatus(DN_SOLVER_FN(potrf, type_prefix)( \
Dcusolver_context.h78 #define POTRF_INSTANCE(T, type_prefix) \ argument
/external/vulkan-headers/registry/
Dconventions.py199 def type_prefix(self): member in ConventionsBase
/external/angle/third_party/vulkan-deps/vulkan-headers/src/registry/
Dconventions.py199 def type_prefix(self): member in ConventionsBase