Home
last modified time | relevance | path

Searched refs:BestCudnnConvAlgorithm (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dgpu_utils.cc271 StatusOr<se::dnn::AlgorithmConfig> BestCudnnConvAlgorithm( in BestCudnnConvAlgorithm() function
294 StatusOr<AutotuneEntry<Op>> BestCudnnConvAlgorithm( in BestCudnnConvAlgorithm() function
318 BestCudnnConvAlgorithm<se::dnn::ConvOp>(
325 BestCudnnConvAlgorithm<se::dnn::FusedConvOp>(
H A Dconv_ops_gpu.cc178 BestCudnnConvAlgorithm<se::dnn::FusedConvOp>( in AutotuneFusedConv()
206 BestCudnnConvAlgorithm<se::dnn::FusedConvOp>( in AutotuneFusedConv()
355 BestCudnnConvAlgorithm<se::dnn::ConvOp>(results, std::move(runners))); in AutotuneUnfusedConv()
379 BestCudnnConvAlgorithm<se::dnn::ConvOp>( in AutotuneUnfusedConv()
432 TF_ASSIGN_OR_RETURN(auto algo_desc, BestCudnnConvAlgorithm(results)); in AutotuneUnfusedConv()
H A Dgpu_utils.h406 StatusOr<se::dnn::AlgorithmConfig> BestCudnnConvAlgorithm(
414 StatusOr<AutotuneEntry<Op>> BestCudnnConvAlgorithm(