Home
last modified time | relevance | path

Searched refs:AlgorithmConfig (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc33 using se::dnn::AlgorithmConfig;
81 AlgorithmConfig algorithm) { in RunGpuConvForward()
101 AlgorithmConfig algorithm) { in RunGpuConvForwardActivation()
154 AlgorithmConfig algorithm) { in RunGpuConvInternalImpl()
200 AlgorithmConfig algorithm) { in RunGpuConvInternalImpl()
224 AlgorithmConfig algorithm = params.algorithm; in RunGpuConvImpl()
227 algorithm = AlgorithmConfig(*options.algo_override); in RunGpuConvImpl()
267 params.algorithm = se::dnn::AlgorithmConfig( in GetGpuConvParams()
Dgpu_conv_runner.h69 se::dnn::AlgorithmConfig algorithm;
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc124 using se::dnn::AlgorithmConfig;
226 AutoTuneSingleton<RnnAutoTuneGroup, CudnnRnnParameters, AlgorithmConfig>;
1036 const AlgorithmConfig algo_config; in ExtractCudnnRNNParamsInfo()
1053 const AlgorithmConfig& algo_config, in CreateRnnDescriptor()
1080 const AlgorithmConfig& algo_config, in GetCachedRnnDescriptor()
1442 AlgorithmConfig algo_config; in Compute()
1449 AlgorithmConfig* output_algo_config, in ComputeAndReturnAlgorithm()
1528 AlgorithmConfig* best_algo_config) { in MaybeAutoTune()
1530 *best_algo_config = AlgorithmConfig(); in MaybeAutoTune()
1596 AlgorithmConfig best_algo_config; in Compute()
[all …]
Dconv_ops_3d.cc204 se::dnn::AlgorithmConfig>
442 using se::dnn::AlgorithmConfig; in launch()
446 AlgorithmConfig algorithm_config; in launch()
486 AlgorithmConfig(profile_algorithm), &profile_result) in launch()
541 AlgorithmConfig(profile_algorithm), &profile_result) in launch()
Dconv_ops_fused_impl.h292 se::dnn::AlgorithmConfig>;
312 se::dnn::AlgorithmConfig* algorithm_config) {
349 launch(se::dnn::AlgorithmConfig(profile_algorithm), allocator_used,
612 const auto launch = [&](se::dnn::AlgorithmConfig algorithm_config,
630 se::dnn::AlgorithmConfig algorithm_config;
Dconv_grad_ops_3d.cc1097 se::dnn::AlgorithmConfig>
1368 using se::dnn::AlgorithmConfig; in Compute()
1371 AlgorithmConfig algorithm_config; in Compute()
1407 AlgorithmConfig(profile_algorithm), &profile_result) in Compute()
1458 AlgorithmConfig(profile_algorithm), &profile_result) in Compute()
1564 se::dnn::AlgorithmConfig>
1857 using se::dnn::AlgorithmConfig; in Compute()
1860 AlgorithmConfig algorithm_config; in Compute()
1882 &scratch_allocator, AlgorithmConfig(profile_algorithm), in Compute()
1920 &scratch_allocator, AlgorithmConfig(profile_algorithm), in Compute()
Dmatmul_op.cc251 se::blas::AlgorithmConfig>
260 using se::blas::AlgorithmConfig; in launch()
294 AlgorithmConfig algorithm_config(kNoAlgorithm); in launch()
Dconv_grad_filter_ops.cc647 se::dnn::AlgorithmConfig>
657 using se::dnn::AlgorithmConfig; in operator ()()
978 AlgorithmConfig algorithm_config; in operator ()()
1015 allocator_used, AlgorithmConfig(profile_algorithm), in operator ()()
1073 &scratch_allocator, AlgorithmConfig(profile_algorithm), in operator ()()
Dconv_ops.cc642 se::dnn::AlgorithmConfig>
652 using se::dnn::AlgorithmConfig; in operator ()()
985 AlgorithmConfig algorithm_config; in operator ()()
1026 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
1084 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
Dconv_grad_input_ops.cc826 se::dnn::AlgorithmConfig>
836 using se::dnn::AlgorithmConfig; in operator ()()
1146 AlgorithmConfig algorithm_config; in operator ()()
1183 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
1239 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
Dgpu_utils.h244 se::dnn::AlgorithmConfig* algo);
Dgpu_utils.cc245 se::dnn::AlgorithmConfig* algo) { in BestCudnnConvAlgorithm()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h837 class AlgorithmConfig {
839 AlgorithmConfig() {}
840 explicit AlgorithmConfig(AlgorithmDesc algorithm) : algorithm_(algorithm) {}
841 AlgorithmConfig(AlgorithmDesc algorithm, size_t scratch_size)
843 AlgorithmConfig(AlgorithmDesc algorithm, AlgorithmDesc algorithm_no_scratch)
855 bool operator==(const AlgorithmConfig& other) const {
860 bool operator!=(const AlgorithmConfig& other) const {
1186 const dnn::AlgorithmConfig& algorithm_config,
1203 const dnn::AlgorithmConfig& algorithm_config,
1223 const dnn::AlgorithmConfig& algorithm_config,
[all …]
Dblas.cc71 string AlgorithmConfig::ToString() const { return absl::StrCat(algorithm_); } in ToString()
Dstream.h323 const dnn::AlgorithmConfig &algorithm_config,
334 const dnn::AlgorithmConfig &algorithm_config,
345 const dnn::AlgorithmConfig &algorithm_config,
356 const dnn::AlgorithmConfig &algorithm_config,
367 const dnn::AlgorithmConfig &algorithm_config,
381 const dnn::AlgorithmConfig &algorithm_config,
395 const dnn::AlgorithmConfig &algorithm_config,
410 const dnn::AlgorithmConfig &algorithm_config,
424 const dnn::AlgorithmConfig &algorithm_config,
438 const dnn::AlgorithmConfig &algorithm_config,
[all …]
Dstream.cc62 string ToVlogString(const dnn::AlgorithmConfig &algo_config) { in ToVlogString()
471 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
509 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
548 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
586 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
624 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
659 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
701 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
743 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
785 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
[all …]
Ddnn.cc250 string AlgorithmConfig::ToString() const { in ToString()
Dstream_executor_pimpl.h408 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config,
Dblas.h148 class AlgorithmConfig {
150 AlgorithmConfig() : algorithm_(kDefaultAlgorithm) {} in AlgorithmConfig() function
151 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig() function
154 bool operator==(const AlgorithmConfig &other) const {
157 bool operator!=(const AlgorithmConfig &other) const {
Dstream_executor_pimpl.cc358 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config, in createRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h54 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config,
295 const dnn::AlgorithmConfig& algorithm_config,
309 const dnn::AlgorithmConfig& algorithm_config,
327 const dnn::AlgorithmConfig& algorithm_config,
341 const dnn::AlgorithmConfig& algorithm_config,
355 const dnn::AlgorithmConfig& algorithm_config,
636 const dnn::AlgorithmConfig& algorithm_config,
709 const dnn::AlgorithmConfig& algorithm_config,
Dcuda_dnn.cc1048 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor()
1076 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in Create()
1183 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config()
1217 dnn::AlgorithmConfig algorithm_config_;
2071 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2599 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionForwardAlgorithm()
2652 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardDataAlgorithm()
2704 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardFilterAlgorithm()
2900 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3225 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolveImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h52 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config,
294 const dnn::AlgorithmConfig& algorithm_config,
308 const dnn::AlgorithmConfig& algorithm_config,
326 const dnn::AlgorithmConfig& algorithm_config,
340 const dnn::AlgorithmConfig& algorithm_config,
786 const dnn::AlgorithmConfig& algorithm_config,
Drocm_dnn.cc2323 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2643 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3445 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3462 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3480 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3497 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()