diff options
| author |  Alan Chiao <alanchiao@google.com> | 2018-07-12 12:16:09 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-12 12:25:21 -0700 |
| commit | e8a65666c6aadbbbd2b19b9322d841b1547dbd35 (patch) | |
| tree | eadeba71d8ad2d4c0c147d59f563d3246a7f6a65 /tensorflow/contrib/lite/kernels/lstm_test.cc | |
| parent | 746a51b76742574d81783a4efe437e1824073d88 (diff) | |
LSTM CHECK_OK on input tensor checks.
PiperOrigin-RevId: 204341675
Diffstat (limited to 'tensorflow/contrib/lite/kernels/lstm_test.cc')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/lstm_test.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/lstm_test.cc b/tensorflow/contrib/lite/kernels/lstm_test.cc index 0b7c56133e..0266f5fe57 100644 --- a/tensorflow/contrib/lite/kernels/lstm_test.cc +++ b/tensorflow/contrib/lite/kernels/lstm_test.cc @@ -13,6 +13,9 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ // Unit test for TFLite LSTM op. +// +// TODO(alanchiao): add unit test with invalid input dimensions for this and its +// variants. #include <memory> #include <vector> |