Searched refs:bw_row_sums (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn.cc | 313 TfLiteTensor* bw_row_sums; in Prepare() local
316 GetTemporarySafe(context, node, /*index=*/kBwRowSums, &bw_row_sums)); in Prepare()
317 bw_row_sums->type = kTfLiteInt32; in Prepare()
318 bw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
320 if (!TfLiteIntArrayEqualsArray(bw_row_sums->dims, 2, bw_row_sums_dims)) { in Prepare()
324 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, bw_row_sums, in Prepare()
529 TfLiteTensor* bw_row_sums, bool* fw_compute_row_sums, in EvalHybrid() argument
583 bw_row_sums_ptr = GetTensorData<int32_t>(bw_row_sums); in EvalHybrid()
815 TfLiteTensor* bw_row_sums; in Eval() local
817 context, GetTemporarySafe(context, node, kBwRowSums, &bw_row_sums)); in Eval()
[all …]
|
| D | bidirectional_sequence_lstm.cc | 931 TfLiteTensor* bw_row_sums; in Prepare() local
933 context, GetTemporarySafe(context, node, kBwRowSums, &bw_row_sums)); in Prepare()
934 bw_row_sums->type = kTfLiteInt32; in Prepare()
935 bw_row_sums->allocation_type = kTfLiteArenaRwPersistent; in Prepare()
937 if (!TfLiteIntArrayEqualsArray(bw_row_sums->dims, 2, bw_row_sums_dims)) { in Prepare()
941 TF_LITE_ENSURE_OK(context, context->ResizeTensor(context, bw_row_sums, in Prepare()
1257 TfLiteTensor* bw_row_sums; in Eval() local
1259 context, GetTemporarySafe(context, node, kBwRowSums, &bw_row_sums)); in Eval()
1261 const int bw_row_sums_size = bw_row_sums->dims->data[0]; in Eval()
1339 GetTemporary(context, node, kOutputStateZeroPoints), bw_row_sums, in Eval()
|