diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-09-07 14:05:22 +0100 |
---|---|---|
committer | finn.williams <finn.williams@arm.com> | 2020-09-08 13:10:16 +0000 |
commit | bbf71a6478d62a8fdc9d153787e26db766efbc16 (patch) | |
tree | 8d660d1a6874e01ee0f75ca38770790bd8085762 /src/armnn | |
parent | ebc071653142efc03751d00870e64db89f6ee3ad (diff) | |
download | armnn-bbf71a6478d62a8fdc9d153787e26db766efbc16.tar.gz |
IVGCVSW-5244 Load-scope dynamic tensor TfLite tests
* Infer tensor shapes at the beginning of Optimize function
* Unit tests
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I511f1228a12ebcad570e42a0c46d461ab9ccdc2c
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/Network.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index dec9468d7b..84997a61e7 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -1037,6 +1037,9 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, // Get the optimized graph Graph& optGraph = optNetObjPtr->GetGraph(); + // Infer the tensor infos for all output slots. Throws an exception on failure + optGraph.InferTensorInfos(); + // Perform optimisation passes using namespace optimizations; Optimizer::Pass(optGraph, MakeOptimizations(SquashEqualPermuteSiblings(), @@ -1053,9 +1056,6 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, PermuteAndBatchToSpaceAsDepthToSpace(), TransposeAndBatchToSpaceAsDepthToSpace())); - // Infer the tensor infos for all output slots. Throws an exception on failure - optGraph.InferTensorInfos(); - // If Fp32 to Fp16 optimization is set convert Fp32 network to Fp16 if (options.m_ReduceFp32ToFp16) { |