Home
last modified time | relevance | path

Searched defs:hidden_state (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc74 const TfLiteTensor* hidden_state; in Prepare() local
204 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalFloat()
235 TfLiteTensor* hidden_state, TfLiteTensor* output, in EvalHybrid()
294 TfLiteTensor* hidden_state = in Eval() local
Dunidirectional_sequence_rnn.cc75 const TfLiteTensor* hidden_state; in Prepare() local
212 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalFloat()
273 TfLiteTensor* hidden_state, TfLiteTensor* output, TfLiteTensor* zero_points, in EvalHybrid()
370 TfLiteTensor* hidden_state = in Eval() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc186 auto hidden_state = builder->create<MulOp>( in LoadForLSTMOp() local