| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 34 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()
|
| D | gpu_conv_runner.h | 67 se::dnn::AlgorithmConfig algorithm;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 124 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 …]
|
| D | conv_grad_input_ops.cc | 43 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 ()()
|
| D | conv_ops_3d.cc | 206 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()
|
| D | conv_ops_fused_impl.h | 349 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;
|
| D | conv_grad_ops_3d.cc | 1099 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()
|
| D | conv_grad_filter_ops.cc | 646 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 ()()
|
| D | conv_ops.cc | 642 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 ()()
|
| D | gpu_utils.h | 245 se::dnn::AlgorithmConfig* algo);
|
| D | gpu_utils.cc | 247 se::dnn::AlgorithmConfig* algo) { in BestCudnnConvAlgorithm()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 814 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 …]
|
| D | blas.cc | 71 std::string AlgorithmConfig::ToString() const { in ToString()
|
| D | stream.h | 334 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()
|
| D | dnn.cc | 255 std::string AlgorithmConfig::ToString() const { in ToString()
|
| D | stream_executor_pimpl.h | 419 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config,
|
| D | blas.h | 178 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 {
|
| D | stream_executor_pimpl.cc | 369 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config, in createRnnDescriptor()
|
| D | stream.cc | 61 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/ |
| D | cuda_dnn.h | 54 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,
|
| D | cuda_dnn.cc | 1065 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/ |
| D | rocm_dnn.h | 88 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,
|
| D | rocm_dnn.cc | 2519 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()
|