vee1e/tflite-micro
1
0
Fork 0
mirror of https://github.com/vee1e/tflite-micro.git synced 2026-09-01 17:57:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
tflite-micro/tensorflow
History
Exact
David Davis 8f1f3b2623
WHILE operator input/output copy fix (#3633) 
@tensorflow/micro

Remove extraneous tensor copy operation after first invocation of condition subgraph.

Move copy of operator inputs to outputs, such that it occurs before the first invocation of the condition subgraph. This preserves the operator inputs when one or more of them is the output of DECODE, and alternate decompression memory is in use. This is because the output of DECODE is for immediate consumption by the next operator in the graph (WHILE), yet it is possible for the WHILE subgraph invocations to share memory with the original DECODE output.

Update the unit test for multiple invocations of the condition and body subgraphs.

When copying tensors between operator inputs/outputs and subgraph inputs/outputs, check if the source and destination tensors share memory.

bug=fixes #3632
2026-07-20 17:58:59 +00:00
..
compiler/mlir/lite Manual sync from sync from github.com/tensorflow/tensorflow (#3616) 2026-07-01 18:13:01 -07:00
lite WHILE operator input/output copy fix (#3633) 2026-07-20 17:58:59 +00:00
BUILD
extra_rules.bzl