Home
last modified time | relevance | path

Searched defs:out_algorithms (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.cc112 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
187 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
193 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
199 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
H A Dstream_executor_pimpl.cc272 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
345 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithms()
357 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
366 Stream* stream, std::vector<blas::AlgorithmType>* out_algorithms) { in GetBlasGemmAlgorithms()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc3134 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
3220 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithms()
3242 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsImmediateMode()
3451 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsFindMode()
3603 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
3611 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
3626 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
H A Drocm_blas.cc653 Stream *stream, std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc5276 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
5314 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
5335 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
5370 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
H A Dcuda_blas.cc1098 Stream *stream, std::vector<blas::AlgorithmType> *out_algorithms) { in DoBlasInternalImpl()