Searched refs:algo_desc (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1876 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnForwardImpl() local
1877 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()
2021 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnBackwardImpl() local
2022 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()
2695 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionForwardAlgorithm() local
2702 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionForwardAlgorithm()
2705 if (!algo_desc.has_value()) { in GetCudnnConvolutionForwardAlgorithm()
2717 algo_desc = dnn::AlgorithmDesc(algo, use_tensor_ops); in GetCudnnConvolutionForwardAlgorithm()
2721 stream, cudnn, input_nd, filter, conv, output_nd, *algo_desc, in GetCudnnConvolutionForwardAlgorithm()
2726 return *algo_desc; in GetCudnnConvolutionForwardAlgorithm()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 533 auto& algo_desc = to_hash.second; in operator ()() local
538 if (algo_desc.has_value()) { in operator ()()
539 hash = Hash64Combine(hash, algo_desc->hash()); in operator ()()
1498 AlgorithmDesc algo_desc(debug_cudnn_rnn_algo_, debug_use_tensor_ops_); in ComputeAndReturnAlgorithm() local
1499 output_algo_config->set_algorithm(algo_desc); in ComputeAndReturnAlgorithm()
2052 const AlgorithmDesc algo_desc(host_reserved_int8(0), host_reserved_int8(1)); in GetAlgorithm() local
2053 algo_config->set_algorithm(algo_desc); in GetAlgorithm()
|