Home
last modified time | relevance | path

Searched refs:CreateBlasLtMatmulPlan (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc341 StreamExecutor::CreateBlasLtMatmulPlan( in CreateBlasLtMatmulPlan() function in stream_executor::StreamExecutor
348 return blas_support->CreateBlasLtMatmulPlan(params); in CreateBlasLtMatmulPlan()
Dstream_executor_pimpl.h403 CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &params);
Dblas.h1458 CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &params) = 0;
2333 CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &params) override; \
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2657 ROCMBlas::CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &p) { in CreateBlasLtMatmulPlan() function in stream_executor::gpu::ROCMBlas
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc3447 CUDABlas::CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &p) { in DoBlasInternalImpl() function in stream_executor::gpu::CUDABlas