Searched refs:fw_output (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn.cc | 347 TfLiteTensor* fw_output; in Prepare() local
349 GetOutputSafe(context, node, kFwOutputTensor, &fw_output)); in Prepare()
356 context, context->ResizeTensor(context, fw_output, fw_output_size_array)); in Prepare()
383 TfLiteTensor* fw_hidden_state, TfLiteTensor* fw_output, in EvalFloat() argument
429 GetTensorData<float>(fw_output) + s * fw_output_step * batch_size; in EvalFloat()
448 ? GetTensorData<float>(fw_output) + fw_num_units in EvalFloat()
464 GetTensorData<float>(fw_output) + b * fw_output_step * max_time; in EvalFloat()
488 ? GetTensorData<float>(fw_output) + in EvalFloat()
525 TfLiteTensor* fw_hidden_state, TfLiteTensor* fw_output, in EvalHybrid() argument
602 GetTensorData<float>(fw_output) + s * fw_output_step * batch_size; in EvalHybrid()
[all …]
|
| D | bidirectional_sequence_rnn_test.cc | 1137 std::vector<float> fw_output = rnn.GetFwOutput(); in TEST() local
1139 EXPECT_EQ(fw_output.size(), bw_output.size()); in TEST()
1141 std::transform(fw_output.begin(), fw_output.end(), bw_output.begin(), in TEST()
1142 fw_output.begin(), std::plus<float>()); in TEST()
1146 const float* rnn_output = fw_output.data() + s * rnn.num_fw_units(); in TEST()
|
| D | bidirectional_sequence_lstm.cc | 532 TfLiteTensor* fw_output; in Prepare() local
534 GetOutputSafe(context, node, kFwOutputTensor, &fw_output)); in Prepare()
556 context->ResizeTensor(context, fw_output, fw_output_size)); in Prepare()
1040 TfLiteTensor* fw_output; in Eval() local
1042 GetOutputSafe(context, node, kFwOutputTensor, &fw_output)); in Eval()
1150 const auto actual_bw_output = params->merge_outputs ? fw_output : bw_output; in Eval()
1192 fw_scratch_buffer, fw_activation_state, fw_cell_state, fw_output); in Eval()
1295 accum_scratch, fw_output, in Eval()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | group_bidirectional_sequence_ops.cc | 38 Operator** fw_output, Operator** bw_output) { in MatchTwoUnpackOps() argument
43 *fw_output = GetOpWithOutput(model, op.inputs[0]); in MatchTwoUnpackOps()
45 if (*fw_output == nullptr || *bw_output == nullptr) { in MatchTwoUnpackOps()
49 if ((*fw_output)->type != OperatorType::kUnpack || in MatchTwoUnpackOps()
60 Operator** fw_output, in MatchDynamicBidirectionalSequenceOutputs() argument
67 *fw_output = op; in MatchDynamicBidirectionalSequenceOutputs()
69 if (*fw_output == nullptr || reverse_output == nullptr) { in MatchDynamicBidirectionalSequenceOutputs()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 4200 AnyTensor:$fw_output,
|