Searched refs:bw_output (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn.cc | 358 TfLiteTensor* bw_output; in Prepare() local
360 context, GetOutputSafe(context, node, kBwOutputTensor, &bw_output)); in Prepare()
365 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, bw_output, in Prepare()
384 TfLiteTensor* bw_hidden_state, TfLiteTensor* bw_output) { in EvalFloat() argument
449 : GetTensorData<float>(bw_output)) + in EvalFloat()
490 : GetTensorData<float>(bw_output) + b * bw_output_step * max_time; in EvalFloat()
527 TfLiteTensor* bw_output, TfLiteTensor* zero_points, in EvalHybrid() argument
628 : GetTensorData<float>(bw_output)) + in EvalHybrid()
681 : GetTensorData<float>(bw_output) + b * bw_output_step * max_time; in EvalHybrid()
755 TfLiteTensor* bw_output = params->merge_outputs in Eval() local
[all …]
|
| D | bidirectional_sequence_lstm.cc | 612 TfLiteTensor* bw_output; in Prepare() local
614 context, GetOutputSafe(context, node, kBwOutputTensor, &bw_output)); in Prepare()
620 context, context->ResizeTensor(context, bw_output, bw_output_size)); in Prepare()
1108 TfLiteTensor* bw_output = params->merge_outputs in Eval() local
1150 const auto actual_bw_output = params->merge_outputs ? fw_output : bw_output; in Eval()
|
| D | bidirectional_sequence_rnn_test.cc | 1138 std::vector<float> bw_output = rnn.GetBwOutput(); 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()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | group_bidirectional_sequence_ops.cc | 38 Operator** fw_output, Operator** bw_output) { in MatchTwoUnpackOps() argument
44 *bw_output = GetOpWithOutput(model, op.inputs[1]); in MatchTwoUnpackOps()
45 if (*fw_output == nullptr || *bw_output == nullptr) { in MatchTwoUnpackOps()
50 (*bw_output)->type != OperatorType::kUnpack) { in MatchTwoUnpackOps()
61 Operator** bw_output) { in MatchDynamicBidirectionalSequenceOutputs() argument
78 *bw_output = reverse_output; in MatchDynamicBidirectionalSequenceOutputs()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 4201 AnyTensor:$bw_output
|