diff options
Diffstat (limited to 'arm_compute/graph/nodes')
-rw-r--r-- | arm_compute/graph/nodes/ActivationLayer.h | 1 | ||||
-rw-r--r-- | arm_compute/graph/nodes/ConvolutionLayer.h | 9 | ||||
-rw-r--r-- | arm_compute/graph/nodes/FullyConnectedLayer.h | 1 | ||||
-rw-r--r-- | arm_compute/graph/nodes/NormalizationLayer.h | 1 | ||||
-rw-r--r-- | arm_compute/graph/nodes/PoolingLayer.h | 1 | ||||
-rw-r--r-- | arm_compute/graph/nodes/SoftmaxLayer.h | 2 |
6 files changed, 6 insertions, 9 deletions
diff --git a/arm_compute/graph/nodes/ActivationLayer.h b/arm_compute/graph/nodes/ActivationLayer.h index ea32dd02a2..efe8112e77 100644 --- a/arm_compute/graph/nodes/ActivationLayer.h +++ b/arm_compute/graph/nodes/ActivationLayer.h @@ -45,7 +45,6 @@ public: // Inherited methods overriden: std::unique_ptr<arm_compute::IFunction> instantiate_node(GraphContext &ctx, ITensor *input, ITensor *output) override; - void print_info() override; private: const ActivationLayerInfo _activation_info; /**< Activation layer info */ diff --git a/arm_compute/graph/nodes/ConvolutionLayer.h b/arm_compute/graph/nodes/ConvolutionLayer.h index 086bf03dfe..04ba3dd6b7 100644 --- a/arm_compute/graph/nodes/ConvolutionLayer.h +++ b/arm_compute/graph/nodes/ConvolutionLayer.h @@ -78,23 +78,26 @@ public: // Inherited methods overriden: std::unique_ptr<arm_compute::IFunction> instantiate_node(GraphContext &ctx, ITensor *input, ITensor *output) override; - void print_info() override; private: /** Instantiates a non-grouped convolution * + * @param[in] input Input tensor + * @param[in] output Output tensor * @param[in] conv_method_hint Hint that specifies which convolution layer method to use * * @return Convolution function */ - std::unique_ptr<arm_compute::IFunction> instantiate_convolution(ConvolutionMethodHint conv_method_hint); + std::unique_ptr<arm_compute::IFunction> instantiate_convolution(ITensor *input, ITensor *output, ConvolutionMethodHint conv_method_hint); /** Instantiates a grouped convolution * + * @param[in] input Input tensor + * @param[in] output Output tensor * @param[in] conv_method_hint Hint that specifies which convolution layer method to use * * @return Grouped Convolution function */ - std::unique_ptr<arm_compute::IFunction> instantiate_grouped_convolution(ConvolutionMethodHint conv_method_hint); + std::unique_ptr<arm_compute::IFunction> instantiate_grouped_convolution(ITensor *input, ITensor *output, ConvolutionMethodHint conv_method_hint); private: unsigned int _conv_width; /**< Convolution width */ diff --git a/arm_compute/graph/nodes/FullyConnectedLayer.h b/arm_compute/graph/nodes/FullyConnectedLayer.h index b05bc96c99..d31e060457 100644 --- a/arm_compute/graph/nodes/FullyConnectedLayer.h +++ b/arm_compute/graph/nodes/FullyConnectedLayer.h @@ -51,7 +51,6 @@ public: // Inherited methods overriden: std::unique_ptr<arm_compute::IFunction> instantiate_node(GraphContext &ctx, ITensor *input, ITensor *output) override; - void print_info() override; // Inherited methods overriden: private: diff --git a/arm_compute/graph/nodes/NormalizationLayer.h b/arm_compute/graph/nodes/NormalizationLayer.h index 52f67d2c31..02efd1cbeb 100644 --- a/arm_compute/graph/nodes/NormalizationLayer.h +++ b/arm_compute/graph/nodes/NormalizationLayer.h @@ -44,7 +44,6 @@ public: // Inherited methods overriden: std::unique_ptr<arm_compute::IFunction> instantiate_node(GraphContext &ctx, ITensor *input, ITensor *output) override; - void print_info() override; private: const NormalizationLayerInfo _norm_info; /**< Normalization layer information */ diff --git a/arm_compute/graph/nodes/PoolingLayer.h b/arm_compute/graph/nodes/PoolingLayer.h index f07800a7b8..87b15d06cb 100644 --- a/arm_compute/graph/nodes/PoolingLayer.h +++ b/arm_compute/graph/nodes/PoolingLayer.h @@ -45,7 +45,6 @@ public: // Inherited methods overriden: std::unique_ptr<arm_compute::IFunction> instantiate_node(GraphContext &ctx, ITensor *input, ITensor *output) override; - void print_info() override; private: const PoolingLayerInfo _pool_info; /**< Pooling layer information */ diff --git a/arm_compute/graph/nodes/SoftmaxLayer.h b/arm_compute/graph/nodes/SoftmaxLayer.h index 1515a0f28a..2e1bd98c8d 100644 --- a/arm_compute/graph/nodes/SoftmaxLayer.h +++ b/arm_compute/graph/nodes/SoftmaxLayer.h @@ -28,7 +28,6 @@ #include "arm_compute/graph/INode.h" #include "arm_compute/graph/Tensor.h" #include "arm_compute/graph/Types.h" - namespace arm_compute { namespace graph @@ -39,7 +38,6 @@ class SoftmaxLayer : public INode public: // Inherited methods overriden: std::unique_ptr<arm_compute::IFunction> instantiate_node(GraphContext &ctx, ITensor *input, ITensor *output) override; - void print_info() override; }; } // namespace graph |