/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 | stream_executor_pimpl.h | 419 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config,
|
D | dnn.cc | 255 std::string AlgorithmConfig::ToString() const { in ToString()
|
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()
|