Searched refs:algo_override (Results 1 – 3 of 3) sorted by relevance
39 absl::optional<se::dnn::AlgorithmDesc> algo_override; member
229 if (options.algo_override.has_value()) { in RunGpuConvImpl()230 algorithm = AlgorithmConfig(*options.algo_override); in RunGpuConvImpl()
472 options.algo_override = alg; in PickBestAlgorithmNoCacheCuda()725 options.algo_override = alg; in PickBestAlgorithmNoCacheRocm()