Include dependency graph for dnnl_threadpool.h:
This graph shows which files directly or indirectly include this file:
Go to the source code of this file.
Functions | |
| dnnl_status_t DNNL_API | dnnl_threadpool_interop_stream_create (dnnl_stream_t *stream, dnnl_engine_t engine, void *threadpool) |
| dnnl_status_t DNNL_API | dnnl_threadpool_interop_stream_get_threadpool (dnnl_stream_t astream, void **threadpool) |
| dnnl_status_t DNNL_API | dnnl_threadpool_interop_sgemm (char transa, char transb, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const float *A, dnnl_dim_t lda, const float *B, dnnl_dim_t ldb, float beta, float *C, dnnl_dim_t ldc, void *threadpool) |
| dnnl_status_t DNNL_API | dnnl_threadpool_interop_gemm_u8s8s32 (char transa, char transb, char offsetc, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const uint8_t *A, dnnl_dim_t lda, uint8_t ao, const int8_t *B, dnnl_dim_t ldb, int8_t bo, float beta, int32_t *C, dnnl_dim_t ldc, const int32_t *co, void *threadpool) |
| dnnl_status_t DNNL_API | dnnl_threadpool_interop_gemm_s8s8s32 (char transa, char transb, char offsetc, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const int8_t *A, dnnl_dim_t lda, int8_t ao, const int8_t *B, dnnl_dim_t ldb, int8_t bo, float beta, int32_t *C, dnnl_dim_t ldc, const int32_t *co, void *threadpool) |
