Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc529 auto& algo_desc = to_hash.second; in operator ()() local
1313 AlgorithmDesc algo_desc(debug_cudnn_rnn_algo_, debug_use_tensor_ops_); in ComputeAndReturnAlgorithm() local
1851 const AlgorithmDesc algo_desc(host_reserved_int8(0), host_reserved_int8(1)); in GetAlgorithm() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1649 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnForwardImpl() local
1806 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnBackwardImpl() local
2348 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionForwardAlgorithm() local
2397 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionBackwardDataAlgorithm() local
2446 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionBackwardFilterAlgorithm() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2616 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in DoPrepareForConvolution() local