Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A 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
H A Dgpu_conv_runner.h161 MaybeFusedConvRunner* runner_cache; member
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_op_handler.cc280 auto* runner_cache = invocation.exec_ctx.request_ctx() in MakeOp() local