Searched refs:pointer_mode (Results 1 – 2 of 2) sorted by relevance
465 cublasLtPointerMode_t CUBLASPointerMode(blas::PointerMode pointer_mode) { in CUBLASPointerMode() argument466 switch (pointer_mode) { in CUBLASPointerMode()557 ScopedCublasPointerMode pointer_mode{blas_}; in DoBlasInternalImpl() local558 if (!pointer_mode.Init(pointer_mode_host ? CUBLAS_POINTER_MODE_HOST in DoBlasInternalImpl()3180 blas::PointerMode pointer_mode, blas::Epilogue epilogue, in DoBlasInternalImpl() argument3196 CUBLASPointerMode(pointer_mode))); in DoBlasInternalImpl()3267 p.pointer_mode, p.epilogue, p.transa, p.transb)); in DoBlasInternalImpl()3549 if ((cuda_plan.params().pointer_mode == blas::PointerMode::kHost) != in DoBlasInternalImpl()
218 PointerMode pointer_mode; member