Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc33 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/
Dcudnn_rnn_ops.cc124 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 …]
Dconv_ops_3d.cc197 se::dnn::AlgorithmConfig>
431 using se::dnn::AlgorithmConfig; in launch()
435 AlgorithmConfig algorithm_config; in launch()
461 AlgorithmConfig(profile_algorithm), &profile_result) in launch()
Dconv_ops_fused_impl.h488 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;
Dmatmul_op.cc248 se::blas::AlgorithmConfig>
257 using se::blas::AlgorithmConfig; in launch()
291 AlgorithmConfig algorithm_config(kNoAlgorithm); in launch()
Dconv_grad_ops_3d.cc1086 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()
Dconv_ops.cc574 se::dnn::AlgorithmConfig>
584 using se::dnn::AlgorithmConfig; in operator ()()
847 AlgorithmConfig algorithm_config; in operator ()()
871 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
Dconv_grad_filter_ops.cc439 se::dnn::AlgorithmConfig>
549 using se::dnn::AlgorithmConfig; in operator ()()
839 AlgorithmConfig algorithm_config; in operator ()()
858 &scratch_allocator, AlgorithmConfig(profile_algorithm), in operator ()()
Dgpu_utils.h182 se::dnn::AlgorithmConfig* algo);
Dgpu_utils.cc111 se::dnn::AlgorithmConfig* algo) { in BestCudnnConvAlgorithm()
Dconv_grad_input_ops.cc573 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/
Ddnn.h798 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 …]
Dblas.cc71 string AlgorithmConfig::ToString() const { return absl::StrCat(algorithm_); } in ToString()
Dstream.h312 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 …]
Dstream.cc62 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 …]
Ddnn.cc224 string AlgorithmConfig::ToString() const { in ToString()
Dstream_executor_pimpl.h411 const dnn::AlgorithmConfig &algorithm_config, float dropout, uint64 seed,
Dblas.h147 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 {
Dstream_executor_pimpl.cc385 const dnn::AlgorithmConfig &algorithm_config, float dropout, uint64 seed, in createRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h54 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,
Dcuda_dnn.cc1010 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/
Drocm_dnn.h52 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,
Drocm_dnn.cc2279 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/
Dfused_conv2d_bias_activation_op.cc344 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()