Searched refs:out_algorithms (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | stream_executor_pimpl.cc | 266 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 …]
|
D | stream_executor_pimpl.h | 366 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);
|
D | dnn.cc | 46 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
|
D | dnn.h | 1328 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);
|
D | blas.h | 1088 std::vector<AlgorithmType> *out_algorithms) = 0; 2070 bool GetBlasGemmAlgorithms(std::vector<blas::AlgorithmType> *out_algorithms) \
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 232 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);
|
D | rocm_dnn.cc | 3085 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 …]
|
D | rocm_blas.cc | 1844 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms() argument
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.cc | 3296 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 …]
|
D | cuda_dnn.h | 205 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;
|
D | cuda_blas.cc | 2193 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()
|