HomeSort by relevance Sort by last modified time
    Searched defs:fw_output (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/lite/kernels/
bidirectional_sequence_rnn_test.cc 1035 std::vector<float> fw_output = rnn.GetFwOutput(); local
    [all...]
bidirectional_sequence_rnn.cc 261 TfLiteTensor* fw_output = GetOutput(context, node, kFwOutputTensor); local
268 context, context->ResizeTensor(context, fw_output, fw_output_size_array));
293 TfLiteTensor* fw_hidden_state, TfLiteTensor* fw_output,
335 fw_output->data.f + s * fw_output_step * batch_size;
353 (params->merge_outputs ? fw_output->data.f + fw_num_units
369 fw_output->data.f + b * fw_output_step * max_time;
391 ? fw_output->data.f + b * bw_output_step * max_time + fw_num_units
425 TfLiteTensor* fw_hidden_state, TfLiteTensor* fw_output,
497 fw_output->data.f + s * fw_output_step * batch_size;
519 (params->merge_outputs ? fw_output->data.f + fw_num_unit
620 TfLiteTensor* fw_output = GetOutput(context, node, kFwOutputTensor); local
    [all...]
bidirectional_sequence_lstm.cc 489 TfLiteTensor* fw_output = GetOutput(context, node, kFwOutputTensor); local
509 context->ResizeTensor(context, fw_output, fw_output_size));
800 TfLiteTensor* fw_output = GetOutput(context, node, kFwOutputTensor); local
    [all...]

Completed in 183 milliseconds