Searched refs:fw_row_sums (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn.cc | 297 TfLiteTensor* fw_row_sums; in Prepare() local
300 GetTemporarySafe(context, node, /*index=*/kFwRowSums, &fw_row_sums)); in Prepare()
301 fw_row_sums->type = kTfLiteInt32; in Prepare()
302 fw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
304 if (!TfLiteIntArrayEqualsArray(fw_row_sums->dims, 2, fw_row_sums_dims)) { in Prepare()
308 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, fw_row_sums, in Prepare()
528 TfLiteTensor* accum_scratch, TfLiteTensor* fw_row_sums, in EvalHybrid() argument
582 fw_row_sums_ptr = GetTensorData<int32_t>(fw_row_sums); in EvalHybrid()
812 TfLiteTensor* fw_row_sums; in Eval() local
814 context, GetTemporarySafe(context, node, kFwRowSums, &fw_row_sums)); in Eval()
[all …]
|
| D | bidirectional_sequence_lstm.cc | 906 TfLiteTensor* fw_row_sums; in Prepare() local
908 context, GetTemporarySafe(context, node, kFwRowSums, &fw_row_sums)); in Prepare()
909 fw_row_sums->type = kTfLiteInt32; in Prepare()
910 fw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
912 if (!TfLiteIntArrayEqualsArray(fw_row_sums->dims, 2, fw_row_sums_dims)) { in Prepare()
916 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, fw_row_sums, in Prepare()
1254 TfLiteTensor* fw_row_sums; in Eval() local
1256 context, GetTemporarySafe(context, node, kFwRowSums, &fw_row_sums)); in Eval()
1260 const int fw_row_sums_size = fw_row_sums->dims->data[0]; in Eval()
1298 GetTemporary(context, node, kOutputStateZeroPoints), fw_row_sums, in Eval()
|