Commit Graph

7 Commits

Author SHA1 Message Date
Mihai Maruseac
1970c2158b [tflite]: Insert nullptr checks when obtaining tensors.
As part of ongoing refactoring, `tflite::GetInput`, `tflite::GetOutput`, `tflite::GetTemporary` and `tflite::GetIntermediates` will return `nullptr` in some cases. Hence, we insert the `nullptr` checks on all usages.

We also insert `nullptr` checks on usages of `tflite::GetVariableInput` and `tflite::GetOptionalInputTensor` but only in the cases where there is no obvious check that `nullptr` is acceptable (that is, we only insert the check for the output of these two functions if the tensor is accessed as if it is always not `nullptr`).

PiperOrigin-RevId: 332521299
Change-Id: I29af455bcb48d0b92e58132d951a3badbd772d56
2020-09-18 14:13:50 -07:00
Karim Nosir
2a96849f47 Update source files with used includes.
PiperOrigin-RevId: 316589177
Change-Id: I0aba0ed1cf9ff478e7890fa53a7749bf844bd26d
2020-06-15 18:42:14 -07:00
Jared Duke
7d40f2c389 Fix bazel TFLM compilation w/ TF_LITE_STATIC_MEMORY
Ensure dynamic string utils aren't compiled when this build define is
present.

PiperOrigin-RevId: 311626904
Change-Id: Ica229bf337019f0f446fdb94aaf42c6b7e7c749e
2020-05-14 16:27:45 -07:00
Christian Sigg
fa3b55e940 Change comment of the external repo #include file origin.
PiperOrigin-RevId: 302466710
Change-Id: I4d87c4272dbedc2b1ccee3ffbdc3be9690b7b480
2020-03-23 10:42:53 -07:00
Jared Duke
4ce6a9b7a4 Rename lite/c/c_api_internal.h to lite/c/common.h
The C types in lite/c/c_api_internal.h are not actually internal,
rather, they are common types used throughout the C++ and C APIs.
Rename the header accordingly.

PiperOrigin-RevId: 282494601
Change-Id: Ia784f35724d774db256ffcbbcdc5bb00e6574417
2019-11-25 21:49:25 -08:00
Sachin Joglekar
498dc69fb5 Adds support for dynamic-shaped outputs from NMSv4 & v5. MaskRCNN has NMS ops where the output size input is not constant.
PiperOrigin-RevId: 269587721
2019-09-17 09:43:55 -07:00
Sachin Joglekar
cb49525a25 Adds builtin ops for NMS & Soft NMS in TFLite
PiperOrigin-RevId: 267028434
2019-09-03 16:42:59 -07:00