Home
last modified time | relevance | path

Searched defs:algo_desc (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc535 auto& algo_desc = to_hash.second; in operator ()() local
1539 AlgorithmDesc algo_desc(debug_cudnn_rnn_algo_, debug_use_tensor_ops_); in ComputeAndReturnAlgorithm() local
2094 const AlgorithmDesc algo_desc(host_reserved_int8(0), host_reserved_int8(1)); in GetAlgorithm() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1962 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnForwardImpl() local
2072 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnForwardImpl() local
2195 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnBackwardImpl() local
2304 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnBackwardImpl() local
2987 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionForwardAlgorithm() local
3050 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionBackwardDataAlgorithm() local
3111 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionBackwardFilterAlgorithm() local