experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove outdated TODOs.

Browse Source 
Dynamic tensor support is already implemented in while.cc

PiperOrigin-RevId: 350792706
Change-Id: I02075ff3ad28349fa0d6c619e4f901cf6b90c148
This commit is contained in:
Yu-Cheng Ling 2021-01-08 10:43:46 -08:00 committed by TensorFlower Gardener
parent d2f068134d
commit 2f1a51263d
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/kernels/while.cc
View File

@ -155,7 +155,6 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
TF_LITE_ENSURE_OK(context, cond_subgraph->AllocateTensors()); TF_LITE_ENSURE_OK(context, cond_subgraph->AllocateTensors());
TfLiteTensor* cond_output = TfLiteTensor* cond_output =
cond_subgraph->tensor(cond_subgraph->outputs()[0]); cond_subgraph->tensor(cond_subgraph->outputs()[0]);
// TODO(ycling): Handle the case the cond subgraph has dynamic tensor outputs.
// This should rarely happens. In most cases the output is static with shape // This should rarely happens. In most cases the output is static with shape
// [1]. However theoretically intermediate tensors in the cond subgraph // [1]. However theoretically intermediate tensors in the cond subgraph
// can be dynamic. // can be dynamic.
@ -181,7 +180,6 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
body_subgraph->tensor(body_subgraph->outputs()[i]); body_subgraph->tensor(body_subgraph->outputs()[i]);
TF_LITE_ENSURE_TYPES_EQ(context, body_input->type, body_output->type); TF_LITE_ENSURE_TYPES_EQ(context, body_input->type, body_output->type);
// TODO(ycling): Support dynamic sized body subgraph.
TF_LITE_ENSURE(context, !IsDynamicTensor(body_output)); TF_LITE_ENSURE(context, !IsDynamicTensor(body_output));
if (!TfLiteIntArrayEqual(body_input->dims, body_output->dims)) { if (!TfLiteIntArrayEqual(body_input->dims, body_output->dims)) {
// If the output shape of the body subgraph is static w.r.t. a fixed // If the output shape of the body subgraph is static w.r.t. a fixed