Searched refs:OpRunner (Results 1 – 7 of 7) sorted by relevance
54 std::unique_ptr<const OpRunner<typename Op::Signature>> runner) { in FromOpRunner()78 port::StatusOr<const OpRunner<typename Op::Signature>*> GetOrCreateRunner( in GetOrCreateRunner()89 port::StatusOr<const OpRunner<typename Op::Signature>*> GetRunner() { in GetRunner()107 std::unique_ptr<const OpRunner<typename Op::Signature>> runner) in LazyOpRunner()112 std::unique_ptr<const OpRunner<typename Op::Signature>> runner_129 static port::StatusOr<std::unique_ptr<const OpRunner<ConvSignature>>>155 static port::StatusOr<std::unique_ptr<const OpRunner<FusedConvSignature>>>
861 class OpRunner; variable870 class OpRunner<void(Args...)> {872 virtual ~OpRunner() {} in ~OpRunner()899 using ConvRunner = OpRunner<ConvSignature>;906 using FusedConvRunner = OpRunner<FusedConvSignature>;
132 StatusOr<std::tuple<const se::dnn::OpRunner<Sig>*, se::DeviceMemoryBase>>134 const se::dnn::OpRunner<Sig>* primary, in AllocateScratchOrFallback()135 const se::dnn::OpRunner<Sig>* no_scratch_fallback) { in AllocateScratchOrFallback()136 const se::dnn::OpRunner<Sig>* selected_runner = primary; in AllocateScratchOrFallback()
297 std::unique_ptr<const se::dnn::OpRunner<typename Op::Signature>>> in BestCudnnConvAlgorithm()321 std::unique_ptr<const se::dnn::OpRunner<se::dnn::ConvSignature>>>328 std::unique_ptr<const se::dnn::OpRunner<se::dnn::FusedConvSignature>>>
311 std::unique_ptr<const se::dnn::OpRunner<typename Op::Signature>> primary, in FromOpRunners()312 std::unique_ptr<const se::dnn::OpRunner<typename Op::Signature>> in FromOpRunners()417 std::unique_ptr<const se::dnn::OpRunner<typename Op::Signature>>>
39 std::vector<std::unique_ptr<const se::dnn::OpRunner<Sig>>>& runners, in AutotuneConvImpl()
4440 : public dnn::OpRunner<void(Args...)> {4595 std::vector<std::unique_ptr<const dnn::OpRunner<Sig>>>* out_runners, in CreateOpRunners()