Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc46 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
61 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
67 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
73 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dstream_executor_pimpl.cc282 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveAlgorithms()
299 std::vector<dnn::ProfileResult> *out_algorithms) { in GetMIOpenConvolveAlgorithms()
310 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetRnnAlgorithms()
320 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardDataAlgorithms()
333 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
345 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2919 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
2937 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithms()
3238 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
3246 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
3261 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Drocm_blas.cc1728 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3336 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
3369 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
3392 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
3424 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dcuda_blas.cc1982 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()