Home
last modified time | relevance | path

Searched refs:out_algorithms (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc325 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveAlgorithms() argument
333 cc_minor, out_algorithms); in GetConvolveAlgorithms()
337 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetRnnAlgorithms() argument
342 return dnn_support->GetRnnAlgorithms(out_algorithms); in GetRnnAlgorithms()
347 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardDataAlgorithms() argument
355 with_winograd_nonfused, cc_major, cc_minor, out_algorithms); in GetConvolveBackwardDataAlgorithms()
360 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardFilterAlgorithms() argument
368 with_winograd_nonfused, cc_major, cc_minor, out_algorithms); in GetConvolveBackwardFilterAlgorithms()
372 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms() argument
377 return blas_support->GetBlasGemmAlgorithms(out_algorithms); in GetBlasGemmAlgorithms()
Dstream_executor_pimpl.h386 std::vector<dnn::AlgorithmDesc> *out_algorithms);
389 bool GetRnnAlgorithms(std::vector<dnn::AlgorithmDesc> *out_algorithms);
394 std::vector<dnn::AlgorithmDesc> *out_algorithms);
400 std::vector<dnn::AlgorithmDesc> *out_algorithms);
403 bool GetBlasGemmAlgorithms(std::vector<blas::AlgorithmType> *out_algorithms);
Ddnn.cc31 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
35 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms() argument
41 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms() argument
47 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms() argument
Ddnn.h1273 std::vector<AlgorithmDesc>* out_algorithms);
1276 virtual bool GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms);
1366 std::vector<AlgorithmDesc>* out_algorithms);
1413 std::vector<AlgorithmDesc>* out_algorithms);
Dblas.h1019 std::vector<AlgorithmType> *out_algorithms) = 0;
1934 bool GetBlasGemmAlgorithms(std::vector<blas::AlgorithmType> *out_algorithms) \
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3044 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
3046 out_algorithms->clear(); in GetConvolveAlgorithms()
3049 out_algorithms->push_back({CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM, in GetConvolveAlgorithms()
3072 out_algorithms->push_back({i, /*use_tensor_ops=*/false}); in GetConvolveAlgorithms()
3074 out_algorithms->push_back({i, /*use_tensor_ops=*/true}); in GetConvolveAlgorithms()
3082 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms() argument
3091 out_algorithms->clear(); in GetRnnAlgorithms()
3093 out_algorithms->push_back({i, /*use_tensor_ops=*/false}); in GetRnnAlgorithms()
3096 out_algorithms->push_back({i, /*use_tensor_ops=*/true}); in GetRnnAlgorithms()
3105 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms() argument
[all …]
Dcuda_dnn.h204 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
207 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
211 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
215 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
Dcuda_blas.cc1982 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms() argument
1989 *out_algorithms = { in GetBlasGemmAlgorithms()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h195 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
198 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
202 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
206 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
Drocm_dnn.cc2950 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
2951 out_algorithms->assign({ in GetConvolveAlgorithms()
2963 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms() argument
2971 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms() argument
2972 out_algorithms->assign({ in GetConvolveBackwardDataAlgorithms()
2986 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms() argument
2987 out_algorithms->assign({ in GetConvolveBackwardFilterAlgorithms()
Drocm_blas.cc1709 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms() argument