Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc31 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
35 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
41 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
47 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dstream_executor_pimpl.cc325 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveAlgorithms()
337 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetRnnAlgorithms()
347 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardDataAlgorithms()
360 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
372 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3044 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
3082 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
3105 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
3140 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dcuda_blas.cc1982 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2950 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
2963 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
2971 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
2986 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Drocm_blas.cc1709 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()