Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.h150 const blas::IBlasLtMatmulAlgorithm *algorithm,
154 port::StatusOr<std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>>
Dcuda_blas.cc3367 class CUDABlasLtMatmulAlgorithm final : public blas::IBlasLtMatmulAlgorithm { in DoBlasInternalImpl()
3460 port::StatusOr<std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>> in DoBlasInternalImpl()
3500 std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>> out_algorithms; in DoBlasInternalImpl()
3516 port::StatusOr<std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>> in DoBlasInternalImpl()
3530 const blas::IBlasLtMatmulAlgorithm *algorithm, DeviceMemoryBase bias) { in DoBlasInternalImpl()
3583 std::unique_ptr<blas::IBlasLtMatmulAlgorithm> unique_remainder_algo; in DoBlasInternalImpl()
3692 const blas::IBlasLtMatmulAlgorithm *algorithm, DeviceMemoryBase bias, in DoBlasInternalImpl()
/external/tensorflow/tensorflow/stream_executor/
Dblas.h204 struct IBlasLtMatmulAlgorithm { struct
205 virtual ~IBlasLtMatmulAlgorithm() {} in ~IBlasLtMatmulAlgorithm() argument
1465 std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>>
1486 const blas::IBlasLtMatmulAlgorithm *algorithm, DeviceMemoryBase bias,
1497 const blas::IBlasLtMatmulAlgorithm *algorithm,
2334 port::StatusOr<std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>> \
2343 const blas::IBlasLtMatmulAlgorithm *algorithm, DeviceMemoryBase bias, \
Dstream_executor_pimpl.h409 port::StatusOr<std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>>
Dstream_executor_pimpl.cc351 port::StatusOr<std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>>
Dstream.cc4331 const blas::IBlasLtMatmulAlgorithm *algorithm, in ThenBlasLtMatmulImpl()
4341 ScratchAllocator *, const blas::IBlasLtMatmulAlgorithm *, in ThenBlasLtMatmulImpl()
4353 ScratchAllocator *, const blas::IBlasLtMatmulAlgorithm *,
4360 ScratchAllocator *, const blas::IBlasLtMatmulAlgorithm *,
4367 ScratchAllocator *, const blas::IBlasLtMatmulAlgorithm *,
4374 ScratchAllocator *, const blas::IBlasLtMatmulAlgorithm *,
4385 const blas::IBlasLtMatmulAlgorithm *,
4396 const blas::IBlasLtMatmulAlgorithm *,
Dstream.h1660 const blas::IBlasLtMatmulAlgorithm *algorithm,
2108 const blas::IBlasLtMatmulAlgorithm *algorithm,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2663 port::StatusOr<std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>>
2677 const blas::IBlasLtMatmulAlgorithm *algorithm, DeviceMemoryBase bias, in DoBlasLtMatmul()