Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc34 TfLiteRNNParams rnn_params; in TEST() local
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc523 if (const auto* rnn_params = op->builtin_options_as_RNNOptions()) { in ParseOpDataTfLite() local
525 ConvertActivation(rnn_params->fused_activation_function()); in ParseOpDataTfLite()
527 rnn_params->asymmetric_quantize_inputs(); in ParseOpDataTfLite()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc1654 CudnnRnnParameters rnn_params( in MaybeAutoTune() local
1661 if (AutoTuneRnnConfigMap::GetInstance()->Find(rnn_params, algo_config)) { in MaybeAutoTune()
1768 AutoTuneRnnConfigMap::GetInstance()->Insert(rnn_params, *algo_config); in MaybeAutoTune()