Searched refs:fw_hidden_state (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dbidirectional_sequence_rnn.cc89 TfLiteTensor* fw_hidden_state = local
91 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, fw_hidden_state,
103 fw_hidden_state->allocation_type = kTfLiteArenaRwPersistent;
132 TfLiteTensor* fw_hidden_state = local
164 fw_hidden_state->data.f + b * fw_num_units;

Completed in 242 milliseconds