/external/tensorflow/tensorflow/core/kernels/ |
D | conv_ops_fused_impl.h | 369 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,
|
D | conv_grad_input_ops.cc | 388 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 ()()
|
D | conv_ops_3d.cc | 505 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()
|
D | conv_ops.cc | 985 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 ()()
|
D | conv_grad_ops_3d.cc | 1424 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()
|
D | conv_grad_filter_ops.cc | 983 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/ |
D | cuda_dnn.cc | 1065 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 …]
|
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,
|
/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() 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/ |
D | dnn.h | 1154 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
|
D | stream.h | 334 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 …]
|
D | stream_executor_pimpl.cc | 369 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()
|
D | stream.cc | 457 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 …]
|
D | stream_executor_pimpl.h | 419 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config,
|