Searched refs:ToAlgorithmDesc (Results 1 – 9 of 9) sorted by relevance
58 TF_ASSIGN_OR_RETURN(auto desc, runner->ToAlgorithmDesc()); in FromOpRunner()103 const AlgorithmDesc& ToAlgorithmDesc() const { return desc_; } in ToAlgorithmDesc() function
888 virtual port::StatusOr<AlgorithmDesc> ToAlgorithmDesc() const = 0;
70 runners.primary->ToAlgorithmDesc().ToProto(); in ConvMapToProto()73 runners.no_scratch_fallback->ToAlgorithmDesc().ToProto(); in ConvMapToProto()
115 se::dnn::AlgorithmDesc ToAlgorithmDesc() const { in ToAlgorithmDesc() function140 return runner->ToAlgorithmDesc(); in operator()
437 auto alg = runner->ToAlgorithmDesc(); in AutotuneOneConvRunner()498 runner->ToAlgorithmDesc().workspace_size(); in AutotuneOneConvRunner()838 TF_ASSIGN_OR_RETURN(auto alg, runners[0]->ToAlgorithmDesc()); in PickBestAlgorithmNoCacheRocm()854 TF_ASSIGN_OR_RETURN(auto alg, runner->ToAlgorithmDesc()); in PickBestAlgorithmNoCacheRocm()
245 algorithm = options.runner_cache->ToAlgorithmDesc(); in RunGpuConvImpl()
61 TF_ASSIGN_OR_RETURN(auto desc, runner->ToAlgorithmDesc()); in AutotuneConvImpl()
4076 port::StatusOr<dnn::AlgorithmDesc> ToAlgorithmDesc() const override { in ToAlgorithmDesc() function in stream_executor::gpu::CudnnLegacyConvRunner4446 port::StatusOr<dnn::AlgorithmDesc> ToAlgorithmDesc() const override { in ToAlgorithmDesc() function in stream_executor::gpu::CudnnExecutionPlanRunner4510 TF_ASSIGN_OR_RETURN(auto desc, ToAlgorithmDesc()); in operator ()()4914 port::StatusOr<dnn::AlgorithmDesc> ToAlgorithmDesc() const override { in ToAlgorithmDesc() function in stream_executor::gpu::CudnnLegacyFusedConvRunner
2977 port::StatusOr<AlgorithmDesc> ToAlgorithmDesc() const override { in ToAlgorithmDesc() function in stream_executor::gpu::RocmConvRunner