Searched refs:IBlasLtMatmulPlan (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_blas.h | 144 const blas::IBlasLtMatmulPlan *plan, 155 GetBlasLtMatmulAlgorithmsInternal(const blas::IBlasLtMatmulPlan *plan,
|
D | cuda_blas.cc | 3258 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/ |
D | stream_executor_pimpl.h | 402 port::StatusOr<std::unique_ptr<blas::IBlasLtMatmulPlan>> 410 GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan,
|
D | blas.h | 196 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, \
|
D | stream_executor_pimpl.cc | 340 port::StatusOr<std::unique_ptr<blas::IBlasLtMatmulPlan>> 352 StreamExecutor::GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan, in GetBlasLtMatmulAlgorithms()
|
D | stream.cc | 4327 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 *,
|
D | stream.h | 1655 const blas::IBlasLtMatmulPlan *plan, 2101 Stream &ThenBlasLtMatmulImpl(const blas::IBlasLtMatmulPlan *plan,
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_blas.cc | 2656 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()
|