Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc265 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveAlgorithms()
298 std::vector<dnn::ProfileResult> *out_algorithms) { in GetMIOpenConvolveAlgorithms()
310 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetRnnAlgorithms()
319 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardDataAlgorithms()
329 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
339 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
Ddnn.cc53 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
81 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
87 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
93 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3079 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
3099 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithms()
3121 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsImmediateMode()
3330 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsFindMode()
3482 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
3490 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
3505 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Drocm_blas.cc1612 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4407 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
4448 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
4474 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
4514 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dcuda_blas.cc2230 std::vector<blas::AlgorithmType> *out_algorithms) { in DoBlasInternalImpl()
3452 std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>> out_algorithms; in DoBlasInternalImpl() local