Home
last modified time | relevance | path

Searched refs:CreateLSTM (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dlstm.h27 GPUOperation CreateLSTM(const OperationDef& definition,
Dlstm.cc88 GPUOperation CreateLSTM(const OperationDef& definition, in CreateLSTM() function
Dlstm_test_util.cc60 GPUOperation operation = CreateLSTM(op_def, env->GetGpuInfo()); in LstmTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.cc51 return absl::make_unique<GPUOperation>(CreateLSTM(op_def, gpu_info)); in SelectLSTM()