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()
64 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
70 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
76 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dstream_executor_pimpl.cc266 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveAlgorithms()
285 std::vector<dnn::ProfileResult> *out_algorithms) { in GetMIOpenConvolveAlgorithms()
297 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetRnnAlgorithms()
307 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardDataAlgorithms()
320 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
332 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3085 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
3105 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithms()
3127 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsImmediateMode()
3336 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsFindMode()
3488 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
3496 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
3511 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Drocm_blas.cc1844 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3296 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
3340 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
3366 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
3405 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dcuda_blas.cc2193 std::vector<blas::AlgorithmType> *out_algorithms) { in DoBlasInternalImpl()
3500 std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>> out_algorithms; in DoBlasInternalImpl() local