Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc34 using se::dnn::AlgorithmConfig;
82 AlgorithmConfig algorithm) { in RunGpuConvForward()
102 AlgorithmConfig algorithm) { in RunGpuConvForwardActivation()
155 AlgorithmConfig algorithm) { in RunGpuConvInternalImpl()
203 AlgorithmConfig algorithm) { in RunGpuConvInternalImpl()
227 AlgorithmConfig algorithm = params.config.algorithm; in RunGpuConvImpl()
230 algorithm = AlgorithmConfig(*options.algo_override); in RunGpuConvImpl()
274 config.algorithm = se::dnn::AlgorithmConfig( in GetGpuConvConfig()
Dgpu_conv_runner.h67 se::dnn::AlgorithmConfig algorithm;
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc124 using se::dnn::AlgorithmConfig;
226 AutoTuneSingleton<RnnAutoTuneGroup, CudnnRnnParameters, AlgorithmConfig>;
1040 const AlgorithmConfig algo_config; in ExtractCudnnRNNParamsInfo()
1057 const AlgorithmConfig& algo_config, in CreateRnnDescriptor()
1084 const AlgorithmConfig& algo_config, in GetCachedRnnDescriptor()
1446 AlgorithmConfig algo_config; in Compute()
1453 AlgorithmConfig* output_algo_config, in ComputeAndReturnAlgorithm()
1532 AlgorithmConfig* best_algo_config) { in MaybeAutoTune()
1534 *best_algo_config = AlgorithmConfig(); in MaybeAutoTune()
1600 AlgorithmConfig best_algo_config; in Compute()
[all …]
Dconv_grad_input_ops.cc43 se::dnn::AlgorithmConfig>
72 using se::dnn::AlgorithmConfig; in operator ()()
388 AlgorithmConfig algorithm_config; in operator ()()
423 AlgorithmConfig(profile_algorithm), &profile_result); in operator ()()
476 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
Dconv_ops_3d.cc206 se::dnn::AlgorithmConfig>
495 using se::dnn::AlgorithmConfig; in launch()
505 AlgorithmConfig algorithm_config; in launch()
543 AlgorithmConfig(profile_algorithm), &profile_result); in launch()
597 AlgorithmConfig(profile_algorithm, in launch()
Dconv_ops_fused_impl.h349 se::dnn::AlgorithmConfig>;
369 se::dnn::AlgorithmConfig* algorithm_config) {
406 launch(se::dnn::AlgorithmConfig(profile_algorithm), allocator_used,
669 const auto launch = [&](se::dnn::AlgorithmConfig algorithm_config,
685 se::dnn::AlgorithmConfig algorithm_config;
Dconv_grad_ops_3d.cc1099 se::dnn::AlgorithmConfig>
1415 using se::dnn::AlgorithmConfig; in Compute()
1424 AlgorithmConfig algorithm_config; in Compute()
1457 AlgorithmConfig(profile_algorithm), &profile_result); in Compute()
1495 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in Compute()
1580 se::dnn::AlgorithmConfig>
1912 using se::dnn::AlgorithmConfig; in Compute()
1921 AlgorithmConfig algorithm_config; in Compute()
1941 AlgorithmConfig(profile_algorithm), &profile_result); in Compute()
1972 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in Compute()
Dconv_grad_filter_ops.cc646 se::dnn::AlgorithmConfig>
656 using se::dnn::AlgorithmConfig; in operator ()()
983 AlgorithmConfig algorithm_config; in operator ()()
1018 AlgorithmConfig(profile_algorithm), &profile_result); in operator ()()
1074 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
Dconv_ops.cc642 se::dnn::AlgorithmConfig>
652 using se::dnn::AlgorithmConfig; in operator ()()
985 AlgorithmConfig algorithm_config; in operator ()()
1030 AlgorithmConfig(profile_algorithm), &profile_result); in operator ()()
1086 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
Dgpu_utils.h245 se::dnn::AlgorithmConfig* algo);
Dgpu_utils.cc247 se::dnn::AlgorithmConfig* algo) { in BestCudnnConvAlgorithm()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h814 class AlgorithmConfig {
816 AlgorithmConfig() {} in AlgorithmConfig() function
817 explicit AlgorithmConfig(AlgorithmDesc algorithm) : algorithm_(algorithm) {} in AlgorithmConfig() function
818 AlgorithmConfig(AlgorithmDesc algorithm, size_t scratch_size) in AlgorithmConfig() function
820 AlgorithmConfig(AlgorithmDesc algorithm, AlgorithmDesc algorithm_no_scratch) in AlgorithmConfig() function
832 bool operator==(const AlgorithmConfig& other) const {
837 bool operator!=(const AlgorithmConfig& other) const {
1154 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
1172 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
1193 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
[all …]
Dblas.cc71 std::string AlgorithmConfig::ToString() const { in ToString()
Dstream.h334 const dnn::AlgorithmConfig &algorithm_config, in ConvolveWithAlgorithm()
365 const dnn::AlgorithmConfig &algorithm_config,
379 const dnn::AlgorithmConfig &algorithm_config,
394 const dnn::AlgorithmConfig &algorithm_config,
408 const dnn::AlgorithmConfig &algorithm_config,
422 const dnn::AlgorithmConfig &algorithm_config,
445 const dnn::AlgorithmConfig &algorithm_config, in ConvolveBackwardDataWithAlgorithm()
477 const dnn::AlgorithmConfig &algorithm_config, in ConvolveBackwardFilterWithAlgorithm()
Dstream_executor_pimpl.h419 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config,
Ddnn.cc255 std::string AlgorithmConfig::ToString() const { in ToString()
Dblas.h178 class AlgorithmConfig {
180 AlgorithmConfig() : algorithm_(kDefaultAlgorithm) {} in AlgorithmConfig() function
181 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig() function
184 bool operator==(const AlgorithmConfig &other) const {
187 bool operator!=(const AlgorithmConfig &other) const {
Dstream_executor_pimpl.cc369 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config, in createRnnDescriptor()
Dstream.cc61 std::string ToVlogString(const dnn::AlgorithmConfig &algo_config) { in ToVlogString()
457 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
488 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
520 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
551 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
582 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
614 /*scratch_allocator=*/nullptr, dnn::AlgorithmConfig(), in ThenConvolve()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h54 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config,
291 const dnn::AlgorithmConfig& algorithm_config,
305 const dnn::AlgorithmConfig& algorithm_config,
321 const dnn::AlgorithmConfig& algorithm_config,
335 const dnn::AlgorithmConfig& algorithm_config,
349 const dnn::AlgorithmConfig& algorithm_config,
629 const dnn::AlgorithmConfig& algorithm_config,
702 const dnn::AlgorithmConfig& algorithm_config,
Dcuda_dnn.cc1065 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor()
1093 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in Create()
1212 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config()
1246 dnn::AlgorithmConfig algorithm_config_;
2074 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2689 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionForwardAlgorithm()
2752 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardDataAlgorithm()
2813 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardFilterAlgorithm()
2990 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3189 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolveImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h88 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config,
329 const dnn::AlgorithmConfig& algorithm_config,
343 const dnn::AlgorithmConfig& algorithm_config,
359 const dnn::AlgorithmConfig& algorithm_config,
373 const dnn::AlgorithmConfig& algorithm_config,
841 const dnn::AlgorithmConfig& algorithm_config,
Drocm_dnn.cc2519 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2850 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3686 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3702 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3719 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3735 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()