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.cc266 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveAlgorithms() argument
274 cc_minor, out_algorithms); in GetConvolveAlgorithms()
285 std::vector<dnn::ProfileResult> *out_algorithms) { in GetMIOpenConvolveAlgorithms() argument
293 convolution_descriptor, scratch_allocator, out_algorithms); in GetMIOpenConvolveAlgorithms()
297 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetRnnAlgorithms() argument
302 return dnn_support->GetRnnAlgorithms(out_algorithms); in GetRnnAlgorithms()
307 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardDataAlgorithms() argument
315 with_winograd_nonfused, cc_major, cc_minor, out_algorithms); in GetConvolveBackwardDataAlgorithms()
320 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardFilterAlgorithms() argument
328 with_winograd_nonfused, cc_major, cc_minor, out_algorithms); in GetConvolveBackwardFilterAlgorithms()
[all …]
Dstream_executor_pimpl.h366 std::vector<dnn::AlgorithmDesc> *out_algorithms);
379 std::vector<dnn::ProfileResult> *out_algorithms);
382 bool GetRnnAlgorithms(std::vector<dnn::AlgorithmDesc> *out_algorithms);
387 std::vector<dnn::AlgorithmDesc> *out_algorithms);
393 std::vector<dnn::AlgorithmDesc> *out_algorithms);
396 bool GetBlasGemmAlgorithms(std::vector<blas::AlgorithmType> *out_algorithms);
Ddnn.cc46 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
64 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms() argument
70 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms() argument
76 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms() argument
Ddnn.h1328 std::vector<AlgorithmDesc>* out_algorithms);
1339 std::vector<ProfileResult>* out_algorithms);
1342 virtual bool GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms);
1432 std::vector<AlgorithmDesc>* out_algorithms);
1479 std::vector<AlgorithmDesc>* out_algorithms);
Dblas.h1088 std::vector<AlgorithmType> *out_algorithms) = 0;
2070 bool GetBlasGemmAlgorithms(std::vector<blas::AlgorithmType> *out_algorithms) \
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h232 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
243 std::vector<dnn::ProfileResult>* out_algorithms) override;
246 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
250 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
254 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
873 std::vector<dnn::ProfileResult>* out_algorithms);
884 std::vector<dnn::ProfileResult>* out_algorithms);
Drocm_dnn.cc3085 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
3086 out_algorithms->assign({ in GetConvolveAlgorithms()
3105 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithms() argument
3111 out_algorithms) in GetMIOpenConvolveAlgorithms()
3116 out_algorithms); in GetMIOpenConvolveAlgorithms()
3127 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsImmediateMode() argument
3235 out_algorithms->emplace_back( in GetMIOpenConvolveAlgorithmsImmediateMode()
3274 out_algorithms->emplace_back( in GetMIOpenConvolveAlgorithmsImmediateMode()
3313 out_algorithms->emplace_back( in GetMIOpenConvolveAlgorithmsImmediateMode()
3336 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsFindMode() argument
[all …]
Drocm_blas.cc1844 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms() argument
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3296 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
3303 out_algorithms->clear(); in GetConvolveAlgorithms()
3331 out_algorithms->push_back({i, /*use_tensor_ops=*/true}); in GetConvolveAlgorithms()
3333 out_algorithms->push_back({i, /*use_tensor_ops=*/false}); in GetConvolveAlgorithms()
3340 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms() argument
3356 out_algorithms->clear(); in GetRnnAlgorithms()
3358 out_algorithms->push_back({i, /*use_tensor_ops=*/false}); in GetRnnAlgorithms()
3359 out_algorithms->push_back({i, /*use_tensor_ops=*/true}); in GetRnnAlgorithms()
3366 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms() argument
3375 out_algorithms->clear(); in GetConvolveBackwardDataAlgorithms()
[all …]
Dcuda_dnn.h205 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
208 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
212 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
216 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
Dcuda_blas.cc2193 std::vector<blas::AlgorithmType> *out_algorithms) { in DoBlasInternalImpl() argument
2200 *out_algorithms = { in DoBlasInternalImpl()
3500 std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>> out_algorithms; in DoBlasInternalImpl() local
3501 out_algorithms.reserve(results.size()); in DoBlasInternalImpl()
3505 out_algorithms.emplace_back(std::make_unique<CUDABlasLtMatmulAlgorithm>( in DoBlasInternalImpl()
3508 return out_algorithms; in DoBlasInternalImpl()