mirror of
https://github.com/vee1e/tflite-micro.git
synced 2026-09-01 17:57:27 +00:00
8f1f3b2623
@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 |
||
|---|---|---|
| .. | ||
| c | ||
| core | ||
| experimental/microfrontend | ||
| kernels | ||
| micro | ||
| python | ||
| schema | ||
| tools | ||
| types | ||
| array.cc | ||
| array.h | ||
| BUILD | ||
| build_def.bzl | ||
| builtin_op_data.h | ||
| builtin_ops.h | ||
| context_util.h | ||
| portable_type_to_tflitetype.h | ||