Searched refs:algo_desc (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1649 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnForwardImpl() local
1650 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()
1806 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnBackwardImpl() local
1807 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()
2348 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionForwardAlgorithm() local
2349 if (!algo_desc.has_value()) { in GetCudnnConvolutionForwardAlgorithm()
2361 algo_desc = dnn::AlgorithmDesc(algo, /*use_tensor_ops=*/true); in GetCudnnConvolutionForwardAlgorithm()
2365 stream, cudnn, input_nd, filter, conv, output_nd, *algo_desc, in GetCudnnConvolutionForwardAlgorithm()
2370 return *algo_desc; in GetCudnnConvolutionForwardAlgorithm()
2373 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionForwardAlgorithm()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 529 auto& algo_desc = to_hash.second; in operator ()() local
534 if (algo_desc.has_value()) { in operator ()()
535 hash = Hash64Combine(hash, algo_desc->hash()); in operator ()()
1313 AlgorithmDesc algo_desc(debug_cudnn_rnn_algo_, debug_use_tensor_ops_); in ComputeAndReturnAlgorithm() local
1314 output_algo_config->set_algorithm(algo_desc); in ComputeAndReturnAlgorithm()
1851 const AlgorithmDesc algo_desc(host_reserved_int8(0), host_reserved_int8(1)); in GetAlgorithm() local
1852 algo_config->set_algorithm(algo_desc); in GetAlgorithm()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2616 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in DoPrepareForConvolution() local
2619 if (!algo_desc.has_value()) { in DoPrepareForConvolution()
2732 *algorithm_desc = *algo_desc; in DoPrepareForConvolution()
|