Home
last modified time | relevance | path

Searched defs:runner_cache (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc716 for (auto& runner_cache : runners) { in PickBestAlgorithmNoCacheCuda() local
736 for (auto& runner_cache : fallback_runners) { in PickBestAlgorithmNoCacheCuda() local
872 MaybeFusedConvRunner runner_cache(std::move(lazy_runner)); in PickBestAlgorithmNoCacheRocm() local
Dgpu_conv_runner.h161 MaybeFusedConvRunner* runner_cache; member
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_op_handler.cc280 auto* runner_cache = invocation.exec_ctx.request_ctx() in MakeOp() local