Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h403 CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &params);
Dstream_executor_pimpl.cc342 const blas::BlasLtMatmulPlanParams &params) { in CreateBlasLtMatmulPlan()
Dblas.h214 struct BlasLtMatmulPlanParams { struct
1458 CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &params) = 0;
2333 CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &params) override; \
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc3260 port::Status init(const blas::BlasLtMatmulPlanParams &p) { in DoBlasInternalImpl()
3325 const blas::BlasLtMatmulPlanParams &params() const { return params_; } in DoBlasInternalImpl()
3344 blas::BlasLtMatmulPlanParams params_; in DoBlasInternalImpl()
3447 CUDABlas::CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &p) { in DoBlasInternalImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2657 ROCMBlas::CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams &p) { in CreateBlasLtMatmulPlan()