Home
last modified time | relevance | path

Searched refs:algorithm_no_scratch (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc249 algorithm.algorithm_no_scratch().has_value() in RunGpuConvImpl()
250 ? absl::StrCat(algorithm.algorithm_no_scratch()->algo_id()) in RunGpuConvImpl()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc261 if (algorithm_no_scratch().has_value()) { in ToString()
262 algo_no_scratch = algorithm_no_scratch()->ToString(); in ToString()
Ddnn.h820 AlgorithmConfig(AlgorithmDesc algorithm, AlgorithmDesc algorithm_no_scratch) in AlgorithmConfig() argument
821 : algorithm_(algorithm), algorithm_no_scratch_(algorithm_no_scratch) {} in AlgorithmConfig()
824 absl::optional<AlgorithmDesc> algorithm_no_scratch() const { in algorithm_no_scratch() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2729 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionForwardAlgorithm()
2791 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionBackwardDataAlgorithm()
2852 algo_desc = algorithm_config.algorithm_no_scratch(); in GetCudnnConvolutionBackwardFilterAlgorithm()