diff options
| author | Freddie Liardet <frederick.liardet@arm.com> | 2021-06-17 13:30:11 +0100 |
|---|---|---|
| committer | frederick.liardet <frederick.liardet@arm.com> | 2021-07-08 14:49:39 +0000 |
| commit | dd23f2af4ee4f41f34499ada9c6cd28c9479c6d3 (patch) | |
| tree | e5111c787cec863f4e188b0b155c158525c63e14 /arm_compute/graph/printers | |
| parent | cfac51c779f9bf05e8b2d386fbfb4022767d1d30 (diff) | |
| download | ComputeLibrary-dd23f2af4ee4f41f34499ada9c6cd28c9479c6d3.tar.gz | |
Add LayerData to all nodes
Create LayerData type to store information about inputs/outputs of
nodes, also adds convolution specific information.
Resolves: COMPMID-4422
Signed-off-by: Freddie Liardet <frederick.liardet@arm.com>
Change-Id: I1c3be1abe2fb5ed085108ab3d34b14a1b8561d38
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5876
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/graph/printers')
| -rw-r--r-- | arm_compute/graph/printers/DotGraphPrinter.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/graph/printers/DotGraphPrinter.h b/arm_compute/graph/printers/DotGraphPrinter.h index d39ddad98c..3c9de8cba9 100644 --- a/arm_compute/graph/printers/DotGraphPrinter.h +++ b/arm_compute/graph/printers/DotGraphPrinter.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2019 Arm Limited. + * Copyright (c) 2018-2019,2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -60,7 +60,7 @@ public: void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) override; void visit(NormalizationLayerNode &n) override; void visit(PoolingLayerNode &n) override; - void default_visit() override; + void default_visit(INode &n) override; private: std::string _info{}; |