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.cc536 auto& algo_desc = to_hash.second; in operator ()() local
1540 AlgorithmDesc algo_desc(debug_cudnn_rnn_algo_, debug_use_tensor_ops_, in ComputeAndReturnAlgorithm() local
2096 const AlgorithmDesc algo_desc(host_reserved_int8(0), host_reserved_int8(1), in GetAlgorithm() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc2022 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnForwardImpl() local
2132 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnForwardImpl() local
2255 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnBackwardImpl() local
2364 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnBackwardImpl() local
3060 std::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionForwardAlgorithm() local
3123 std::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionBackwardDataAlgorithm() local
3184 std::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionBackwardFilterAlgorithm() local
5802 dnn::AlgorithmDesc algo_desc; in DoFusedConvolve() local