Searched refs:SetCublasLtAttr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 3073 inline port::Status SetCublasLtAttr(cublasLtMatrixLayout_t handle, in DoBlasInternalImpl() function
3088 inline port::Status SetCublasLtAttr(cublasLtMatmulAlgo_t *handle, in DoBlasInternalImpl() function
3103 inline port::Status SetCublasLtAttr(cublasLtMatmulPreference_t handle, in DoBlasInternalImpl() function
3139 inline port::Status SetCublasLtAttr(cublasLtMatmulDesc_t handle, in DoBlasInternalImpl() function
3195 SE_RETURN_IF_ERROR(SetCublasLtAttr(desc, CUBLASLT_MATMUL_DESC_POINTER_MODE, in DoBlasInternalImpl()
3197 SE_RETURN_IF_ERROR(SetCublasLtAttr(desc, CUBLASLT_MATMUL_DESC_EPILOGUE, in DoBlasInternalImpl()
3199 SE_RETURN_IF_ERROR(SetCublasLtAttr(desc, CUBLASLT_MATMUL_DESC_TRANSA, in DoBlasInternalImpl()
3201 SE_RETURN_IF_ERROR(SetCublasLtAttr(desc, CUBLASLT_MATMUL_DESC_TRANSB, in DoBlasInternalImpl()
3219 SetCublasLtAttr(desc, CUBLASLT_MATRIX_LAYOUT_BATCH_COUNT, batch_count)); in DoBlasInternalImpl()
3220 SE_RETURN_IF_ERROR(SetCublasLtAttr( in DoBlasInternalImpl()
[all …]
|