Home
last modified time | relevance | path

Searched refs:out_runners (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h250 std::vector<std::unique_ptr<const dnn::ConvRunner>>* out_runners)
Drocm_dnn.cc3155 std::vector<std::unique_ptr<const dnn::ConvRunner>>* out_runners) { in GetConvolveRunners() argument
3179 out_runners->push_back(std::move(runner)); in GetConvolveRunners()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc4595 std::vector<std::unique_ptr<const dnn::OpRunner<Sig>>>* out_runners, in CreateOpRunners() argument
4656 out_runners->clear(); in CreateOpRunners()
4693 out_runners->push_back(std::make_unique<CudnnExecutionPlanRunner<Sig>>( in CreateOpRunners()
4702 VLOG(4) << "\nReturned execution plans size: " << out_runners->size(); in CreateOpRunners()