Home
last modified time | relevance | path

Searched refs:algorithm_config (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h369 se::dnn::AlgorithmConfig* algorithm_config) {
371 if (AutoTuneFusedConv::GetInstance()->Find(params, algorithm_config)) {
427 TF_RETURN_IF_ERROR(BestCudnnConvAlgorithm(results, algorithm_config));
428 AutoTuneFusedConv::GetInstance()->Insert(params, *algorithm_config);
669 const auto launch = [&](se::dnn::AlgorithmConfig algorithm_config,
682 scratch_allocator, algorithm_config, profile_result);
685 se::dnn::AlgorithmConfig algorithm_config;
696 &algorithm_config);
701 Status cudnn_launch_status = launch(algorithm_config, &scratch_allocator,
Dconv_grad_input_ops.cc388 AlgorithmConfig algorithm_config; in operator ()() local
390 conv_parameters, &algorithm_config)) { in operator ()()
496 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in operator ()()
498 algorithm_config); in operator ()()
502 input_desc, &in_backprop_ptr, &scratch_allocator, algorithm_config, in operator ()()
Dconv_ops_3d.cc505 AlgorithmConfig algorithm_config; in launch() local
508 conv_parameters, &algorithm_config)) { in launch()
620 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in launch()
621 AutoTuneConv3d::GetInstance()->Insert(conv_parameters, algorithm_config); in launch()
627 &output_ptr, &scratch_allocator, algorithm_config, nullptr); in launch()
Dconv_ops.cc985 AlgorithmConfig algorithm_config; in operator ()() local
993 !AutoTuneConv::GetInstance()->Find(conv_parameters, &algorithm_config)) { in operator ()()
1106 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in operator ()()
1107 AutoTuneConv::GetInstance()->Insert(conv_parameters, algorithm_config); in operator ()()
1111 << algorithm_config.algorithm()->algo_id(); in operator ()()
1113 << algorithm_config.algorithm()->tensor_ops_enabled(); in operator ()()
1118 &output_ptr, &scratch_allocator, algorithm_config, nullptr); in operator ()()
Dconv_grad_ops_3d.cc1424 AlgorithmConfig algorithm_config; in Compute() local
1426 conv_parameters, &algorithm_config)) { in Compute()
1517 BestCudnnConvAlgorithm(results, &algorithm_config)); in Compute()
1519 algorithm_config); in Compute()
1525 input_desc, &in_backprop_ptr, &scratch_allocator, algorithm_config, in Compute()
1921 AlgorithmConfig algorithm_config; in Compute() local
1923 conv_parameters, &algorithm_config)) { in Compute()
1993 Status s = BestCudnnConvAlgorithm(results, &algorithm_config); in Compute()
2012 algorithm_config); in Compute()
2018 filter_desc, &filter_backprop_ptr, &scratch_allocator, algorithm_config, in Compute()
Dconv_grad_filter_ops.cc983 AlgorithmConfig algorithm_config; in operator ()() local
985 conv_parameters, &algorithm_config)) { in operator ()()
1095 OP_REQUIRES_OK(ctx, BestCudnnConvAlgorithm(results, &algorithm_config)); in operator ()()
1097 algorithm_config); in operator ()()
1102 filter_desc, &filter_backprop_ptr, &scratch_allocator, algorithm_config, in operator ()()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1065 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor() argument
1075 rnn_algo_(ToCudnnRNNAlgo(algorithm_config.algorithm())), in CudnnRnnDescriptor()
1081 algorithm_config_(algorithm_config), in CudnnRnnDescriptor()
1093 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in Create() argument
1100 cudnnRNNAlgo_t rnn_algo = ToCudnnRNNAlgo(algorithm_config.algorithm()); in Create()
1118 algorithm_config.algorithm().has_value() in Create()
1119 ? algorithm_config.algorithm()->tensor_ops_enabled() in Create()
1197 data_type, compute_type, algorithm_config, in Create()
1212 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config() function in stream_executor::gpu::CudnnRnnDescriptor
1876 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnForwardImpl()
[all …]
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,
/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() argument
2850 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution() argument
2854 algorithm_config.algorithm(); in DoPrepareForConvolution()
2861 assert(algorithm_config.scratch_size().has_value()); in DoPrepareForConvolution()
2863 size_t scratch_memory_size = *(algorithm_config.scratch_size()); in DoPrepareForConvolution()
3686 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
3702 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
3719 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
3735 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1154 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
1172 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
1193 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
1212 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
1250 const AlgorithmConfig& algorithm_config, in PrepareForConvolution() argument
1256 output_data, convolution_descriptor, algorithm_config, in PrepareForConvolution()
2115 const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor() argument
2685 const AlgorithmConfig& algorithm_config, in DoPrepareForConvolution() argument
Dstream.h334 const dnn::AlgorithmConfig &algorithm_config, in ConvolveWithAlgorithm() argument
342 convolution_descriptor, algorithm_config, scratch_allocator, 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() argument
454 algorithm_config, scratch_allocator, &algorithm_desc, in ConvolveBackwardDataWithAlgorithm()
477 const dnn::AlgorithmConfig &algorithm_config, in ConvolveBackwardFilterWithAlgorithm() argument
[all …]
Dstream_executor_pimpl.cc369 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config, in createRnnDescriptor() argument
379 direction_mode, rnn_mode, data_type, algorithm_config, dropout, seed, in createRnnDescriptor()
Dstream.cc457 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm() argument
464 PARAM(algorithm_config)); in FusedConvolveWithAlgorithm()
471 output_descriptor, output, scratch_allocator, algorithm_config, in FusedConvolveWithAlgorithm()
488 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm() argument
495 PARAM(algorithm_config)); in FusedConvolveWithAlgorithm()
502 output_descriptor, output, scratch_allocator, algorithm_config, in FusedConvolveWithAlgorithm()
520 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm() argument
527 PARAM(output_descriptor), PARAM(output), PARAM(algorithm_config)); in FusedConvolveWithAlgorithm()
534 output_descriptor, output, scratch_allocator, algorithm_config, in FusedConvolveWithAlgorithm()
551 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm() argument
[all …]
Dstream_executor_pimpl.h419 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config,