Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.h36 absl::optional<se::dnn::AlgorithmDesc> algo_override; member
Dcudnn_conv_runner.cc412 if (options.algo_override) { in RunCudnnConv()
413 params.algorithm = AlgorithmConfig(*options.algo_override); in RunCudnnConv()
Dcudnn_conv_algorithm_picker.cc244 options.algo_override = alg; in PickBestAlgorithm()