Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream_executor_pimpl.cc272 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
283 GetDeviceDescription().cuda_compute_capability(), out_algorithms); in GetConvolveAlgorithms()
286 GetDeviceDescription().cuda_compute_capability(), out_algorithms); in GetConvolveAlgorithms()
289 GetDeviceDescription().cuda_compute_capability(), out_algorithms); in GetConvolveAlgorithms()
345 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithms() argument
353 convolution_descriptor, scratch_allocator, out_algorithms); in GetMIOpenConvolveAlgorithms()
357 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms() argument
362 return dnn_support->GetRnnAlgorithms(out_algorithms); in GetRnnAlgorithms()
366 Stream* stream, std::vector<blas::AlgorithmType>* out_algorithms) { in GetBlasGemmAlgorithms() argument
371 return blas_support->GetBlasGemmAlgorithms(stream, out_algorithms); in GetBlasGemmAlgorithms()
Dstream_executor_pimpl.h369 std::vector<dnn::AlgorithmDesc>* out_algorithms);
408 std::vector<dnn::ProfileResult>* out_algorithms);
411 bool GetRnnAlgorithms(std::vector<dnn::AlgorithmDesc>* out_algorithms);
415 std::vector<blas::AlgorithmType>* out_algorithms);
Ddnn.cc112 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
187 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms() argument
193 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms() argument
199 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms() argument
Ddnn.h1387 std::vector<AlgorithmDesc>* out_algorithms);
1445 std::vector<ProfileResult>* out_algorithms);
1448 virtual bool GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms);
1498 std::vector<AlgorithmDesc>* out_algorithms);
1504 std::vector<AlgorithmDesc>* out_algorithms);
Dblas.h391 Stream *stream, std::vector<AlgorithmType> *out_algorithms) = 0;
683 std::vector<blas::AlgorithmType> *out_algorithms) \
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h238 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
270 std::vector<dnn::ProfileResult>* out_algorithms) override;
273 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
277 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
281 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
801 std::vector<dnn::ProfileResult>* out_algorithms);
812 std::vector<dnn::ProfileResult>* out_algorithms);
Drocm_dnn.cc3134 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
3135 out_algorithms->assign({ in GetConvolveAlgorithms()
3220 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithms() argument
3226 out_algorithms) in GetMIOpenConvolveAlgorithms()
3231 out_algorithms); in GetMIOpenConvolveAlgorithms()
3242 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsImmediateMode() argument
3350 out_algorithms->emplace_back( in GetMIOpenConvolveAlgorithmsImmediateMode()
3389 out_algorithms->emplace_back( in GetMIOpenConvolveAlgorithmsImmediateMode()
3428 out_algorithms->emplace_back( in GetMIOpenConvolveAlgorithmsImmediateMode()
3451 std::vector<dnn::ProfileResult>* out_algorithms) { in GetMIOpenConvolveAlgorithmsFindMode() argument
[all …]
Drocm_blas.cc653 Stream *stream, std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms() argument
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h218 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
269 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
273 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
277 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
Dcuda_dnn.cc5276 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms() argument
5281 out_algorithms->clear(); in GetConvolveAlgorithms()
5305 out_algorithms->push_back({i, /*use_tensor_ops=*/true}); in GetConvolveAlgorithms()
5307 out_algorithms->push_back({i, /*use_tensor_ops=*/false}); in GetConvolveAlgorithms()
5314 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms() argument
5325 out_algorithms->clear(); in GetRnnAlgorithms()
5327 out_algorithms->push_back({i, /*use_tensor_ops=*/false}); in GetRnnAlgorithms()
5328 out_algorithms->push_back({i, /*use_tensor_ops=*/true}); in GetRnnAlgorithms()
5335 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms() argument
5340 out_algorithms->clear(); in GetConvolveBackwardDataAlgorithms()
[all …]
Dcuda_blas.cc1098 Stream *stream, std::vector<blas::AlgorithmType> *out_algorithms) { in DoBlasInternalImpl() argument
1110 *out_algorithms = { in DoBlasInternalImpl()
1115 *out_algorithms = { in DoBlasInternalImpl()