/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_conv_runner.cc | 33 using se::dnn::AlgorithmConfig; 72 se::dnn::AlgorithmConfig algorithm; 127 AlgorithmConfig algorithm = params.algorithm; in RunCudnnConvImpl() 337 params.algorithm = se::dnn::AlgorithmConfig(se::dnn::AlgorithmDesc( in GetCudnnConvParams() 413 params.algorithm = AlgorithmConfig(*options.algo_override); in RunCudnnConv()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_rnn_ops.cc | 124 using se::dnn::AlgorithmConfig; 226 AutoTuneSingleton<RnnAutoTuneGroup, CudnnRnnParameters, AlgorithmConfig>; 963 const AlgorithmConfig algo_config; in ExtractCudnnRNNParamsInfo() 979 const AlgorithmConfig& algo_config, in CreateRnnDescriptor() 1004 const AlgorithmConfig& algo_config, in GetCachedRnnDescriptor() 1265 AlgorithmConfig algo_config; in Compute() 1272 AlgorithmConfig* output_algo_config, in ComputeAndReturnAlgorithm() 1347 AlgorithmConfig* best_algo_config) { in MaybeAutoTune() 1349 *best_algo_config = AlgorithmConfig(); in MaybeAutoTune() 1414 AlgorithmConfig best_algo_config; in Compute() [all …]
|
D | conv_ops_3d.cc | 197 se::dnn::AlgorithmConfig> 431 using se::dnn::AlgorithmConfig; in launch() 435 AlgorithmConfig algorithm_config; in launch() 461 AlgorithmConfig(profile_algorithm), &profile_result) in launch()
|
D | conv_ops_fused_impl.h | 488 se::dnn::AlgorithmConfig>; 507 se::dnn::AlgorithmConfig* algorithm_config) { 530 launch(se::dnn::AlgorithmConfig(profile_algorithm), &scratch_allocator, 776 const auto launch = [&](se::dnn::AlgorithmConfig algorithm_config, 793 se::dnn::AlgorithmConfig algorithm_config;
|
D | matmul_op.cc | 248 se::blas::AlgorithmConfig> 257 using se::blas::AlgorithmConfig; in launch() 291 AlgorithmConfig algorithm_config(kNoAlgorithm); in launch()
|
D | conv_grad_ops_3d.cc | 1086 se::dnn::AlgorithmConfig> 1354 using se::dnn::AlgorithmConfig; in Compute() 1357 AlgorithmConfig algorithm_config; in Compute() 1378 AlgorithmConfig(profile_algorithm), &profile_result) in Compute() 1469 se::dnn::AlgorithmConfig> 1760 using se::dnn::AlgorithmConfig; in Compute() 1763 AlgorithmConfig algorithm_config; in Compute() 1784 &scratch_allocator, AlgorithmConfig(profile_algorithm), in Compute()
|
D | conv_ops.cc | 574 se::dnn::AlgorithmConfig> 584 using se::dnn::AlgorithmConfig; in operator ()() 847 AlgorithmConfig algorithm_config; in operator ()() 871 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
|
D | conv_grad_filter_ops.cc | 439 se::dnn::AlgorithmConfig> 549 using se::dnn::AlgorithmConfig; in operator ()() 839 AlgorithmConfig algorithm_config; in operator ()() 858 &scratch_allocator, AlgorithmConfig(profile_algorithm), in operator ()()
|
D | gpu_utils.h | 182 se::dnn::AlgorithmConfig* algo);
|
D | gpu_utils.cc | 111 se::dnn::AlgorithmConfig* algo) { in BestCudnnConvAlgorithm()
|
D | conv_grad_input_ops.cc | 573 se::dnn::AlgorithmConfig> 686 using se::dnn::AlgorithmConfig; in operator ()() 951 AlgorithmConfig algorithm_config; in operator ()() 970 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 798 class AlgorithmConfig { 800 AlgorithmConfig() {} 801 explicit AlgorithmConfig(AlgorithmDesc algorithm) : algorithm_(algorithm) {} 802 AlgorithmConfig(AlgorithmDesc algorithm, AlgorithmDesc algorithm_no_scratch) 812 bool operator==(const AlgorithmConfig& other) const { 816 bool operator!=(const AlgorithmConfig& other) const { 1126 const dnn::AlgorithmConfig& algorithm_config, 1143 const dnn::AlgorithmConfig& algorithm_config, 1163 const dnn::AlgorithmConfig& algorithm_config, 1181 const dnn::AlgorithmConfig& algorithm_config, [all …]
|
D | blas.cc | 71 string AlgorithmConfig::ToString() const { return absl::StrCat(algorithm_); } in ToString()
|
D | stream.h | 312 const dnn::AlgorithmConfig &algorithm_config, 323 const dnn::AlgorithmConfig &algorithm_config, 334 const dnn::AlgorithmConfig &algorithm_config, 348 const dnn::AlgorithmConfig &algorithm_config, 362 const dnn::AlgorithmConfig &algorithm_config, 377 const dnn::AlgorithmConfig &algorithm_config, 391 const dnn::AlgorithmConfig &algorithm_config, 413 const dnn::AlgorithmConfig &algorithm_config, 425 const dnn::AlgorithmConfig &algorithm_config, 437 const dnn::AlgorithmConfig &algorithm_config, [all …]
|
D | stream.cc | 62 string ToVlogString(const dnn::AlgorithmConfig &algo_config) { in ToVlogString() 451 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm() 489 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm() 528 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm() 566 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm() 601 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm() 643 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm() 685 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm() 730 /*scratch_allocator=*/nullptr, dnn::AlgorithmConfig(), in ThenConvolve() 830 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveBackwardDataWithAlgorithm() [all …]
|
D | dnn.cc | 224 string AlgorithmConfig::ToString() const { in ToString()
|
D | stream_executor_pimpl.h | 411 const dnn::AlgorithmConfig &algorithm_config, float dropout, uint64 seed,
|
D | blas.h | 147 class AlgorithmConfig { 149 AlgorithmConfig() : algorithm_(kDefaultAlgorithm) {} in AlgorithmConfig() function 150 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig() function 153 bool operator==(const AlgorithmConfig &other) const { 156 bool operator!=(const AlgorithmConfig &other) const {
|
D | stream_executor_pimpl.cc | 385 const dnn::AlgorithmConfig &algorithm_config, float dropout, uint64 seed, in createRnnDescriptor()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 54 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, 284 const dnn::AlgorithmConfig& algorithm_config, 298 const dnn::AlgorithmConfig& algorithm_config, 316 const dnn::AlgorithmConfig& algorithm_config, 330 const dnn::AlgorithmConfig& algorithm_config, 595 const dnn::AlgorithmConfig& algorithm_config, 659 const dnn::AlgorithmConfig& algorithm_config,
|
D | cuda_dnn.cc | 1010 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor() 1037 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in Create() 1117 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config() 1148 dnn::AlgorithmConfig algorithm_config_; 1820 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in createRnnDescriptor() 2343 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionForwardAlgorithm() 2392 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardDataAlgorithm() 2441 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardFilterAlgorithm() 2632 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution() 2934 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolveImpl() [all …]
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 52 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, 275 const dnn::AlgorithmConfig& algorithm_config, 289 const dnn::AlgorithmConfig& algorithm_config, 307 const dnn::AlgorithmConfig& algorithm_config, 321 const dnn::AlgorithmConfig& algorithm_config, 766 const dnn::AlgorithmConfig& algorithm_config,
|
D | rocm_dnn.cc | 2279 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in createRnnDescriptor() 2598 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution() 3156 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() 3173 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() 3191 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() 3208 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
|
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
D | fused_conv2d_bias_activation_op.cc | 344 se::dnn::AlgorithmConfig* algo) { in BestCudnnConvAlgorithm() 392 dnn::AlgorithmConfig> 692 dnn::AlgorithmConfig algorithm_config; in launch() 727 dnn::AlgorithmConfig(profile_algorithm), &profile_result) in launch()
|