Searched refs:runner_cache (Results 1 – 6 of 6) sorted by relevance
148 auto runner_cache, in GetAlgorithms()151 result.emplace_back(std::move(runner_cache)); in GetAlgorithms()173 auto runner_cache, in GetAlgorithms()176 result.emplace_back(std::move(runner_cache)); in GetAlgorithms()516 options.runner_cache = runner; in AutotuneOneConvRunner()716 for (auto& runner_cache : runners) { in PickBestAlgorithmNoCacheCuda() local720 stream, &runner_cache, operand_buffers, result_buffer, in PickBestAlgorithmNoCacheCuda()736 for (auto& runner_cache : fallback_runners) { in PickBestAlgorithmNoCacheCuda() local740 stream, &runner_cache, operand_buffers, in PickBestAlgorithmNoCacheCuda()872 MaybeFusedConvRunner runner_cache(std::move(lazy_runner)); in PickBestAlgorithmNoCacheRocm() local[all …]
70 opts.runner_cache = &GetOrCreateRunner(params.stream); in ExecuteOnStream()
94 options.runner_cache->AsConvRunner(); in RunGpuConvUnfused()142 options.runner_cache->AsFusedConvRunner(); in RunGpuConvForwardActivation()244 if (options.runner_cache) { in RunGpuConvImpl()245 algorithm = options.runner_cache->ToAlgorithmDesc(); in RunGpuConvImpl()
161 MaybeFusedConvRunner* runner_cache; member
855 opts.runner_cache = &runner; in operator ()()
280 auto* runner_cache = invocation.exec_ctx.request_ctx() in MakeOp() local285 auto kernel_runner_or_status = runner_cache->GetOrCreate( in MakeOp()