Home
last modified time | relevance | path

Searched refs:IBlasLtMatmulPlan (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.h144 const blas::IBlasLtMatmulPlan *plan,
155 GetBlasLtMatmulAlgorithmsInternal(const blas::IBlasLtMatmulPlan *plan,
Dcuda_blas.cc3258 class CUDABlasLtMatmulPlan final : public blas::IBlasLtMatmulPlan { in DoBlasInternalImpl()
3392 const blas::IBlasLtMatmulPlan *plan, size_t max_workspace_bytes) { in DoBlasInternalImpl()
3446 port::StatusOr<std::unique_ptr<blas::IBlasLtMatmulPlan>> in DoBlasInternalImpl()
3451 return static_cast<std::unique_ptr<blas::IBlasLtMatmulPlan>>( in DoBlasInternalImpl()
3461 CUDABlas::GetBlasLtMatmulAlgorithmsInternal(const blas::IBlasLtMatmulPlan *plan, in DoBlasInternalImpl()
3517 CUDABlas::GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan, in DoBlasInternalImpl()
3526 Stream *stream, bool err_on_failure, const blas::IBlasLtMatmulPlan *plan, in DoBlasInternalImpl()
3688 Stream *stream, const blas::IBlasLtMatmulPlan *plan, in DoBlasInternalImpl()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h402 port::StatusOr<std::unique_ptr<blas::IBlasLtMatmulPlan>>
410 GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan,
Dblas.h196 struct IBlasLtMatmulPlan { struct
201 virtual ~IBlasLtMatmulPlan() {} in ~IBlasLtMatmulPlan() argument
1457 virtual port::StatusOr<std::unique_ptr<blas::IBlasLtMatmulPlan>>
1466 GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan,
1482 Stream *stream, const blas::IBlasLtMatmulPlan *plan,
1490 bool DoBlasLtMatmul(Stream *stream, const blas::IBlasLtMatmulPlan *plan,
2332 port::StatusOr<std::unique_ptr<blas::IBlasLtMatmulPlan>> \
2335 GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan, \
2339 Stream *stream, const blas::IBlasLtMatmulPlan *plan, \
Dstream_executor_pimpl.cc340 port::StatusOr<std::unique_ptr<blas::IBlasLtMatmulPlan>>
352 StreamExecutor::GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan, in GetBlasLtMatmulAlgorithms()
Dstream.cc4327 const blas::IBlasLtMatmulPlan *plan, const HostOrDeviceScalar<CType> &alpha, in ThenBlasLtMatmulImpl()
4338 const blas::IBlasLtMatmulPlan *, const HostOrDeviceScalar<CType> &, in ThenBlasLtMatmulImpl()
4350 const blas::IBlasLtMatmulPlan *, const HostOrDeviceScalar<int32> &,
4357 const blas::IBlasLtMatmulPlan *, const HostOrDeviceScalar<Eigen::half> &,
4364 const blas::IBlasLtMatmulPlan *, const HostOrDeviceScalar<float> &,
4371 const blas::IBlasLtMatmulPlan *, const HostOrDeviceScalar<double> &,
4379 const blas::IBlasLtMatmulPlan *,
4390 const blas::IBlasLtMatmulPlan *,
Dstream.h1655 const blas::IBlasLtMatmulPlan *plan,
2101 Stream &ThenBlasLtMatmulImpl(const blas::IBlasLtMatmulPlan *plan,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2656 port::StatusOr<std::unique_ptr<blas::IBlasLtMatmulPlan>>
2664 ROCMBlas::GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan, in GetBlasLtMatmulAlgorithms()
2673 Stream *stream, const blas::IBlasLtMatmulPlan *plan, in DoBlasLtMatmul()